c++樹狀數組2模板

原題

區間修改，單點求值，那麼都需要logn的複雜度，將修改的效果添加到大節點上，向下遍歷樹，這樣最後求單點時向上遍歷樹。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<algorithm>
#define in(x) scanf("%d",&x);
using namespace std;
int n,m,a[500001],c[500001];
void addhead(int pos,int v)
{
    for(int i=pos;i>0;i-=i&(-i))
    c[i]+=v;
}
int sum(int pos)
{
    int ans=a[pos];
    for(int i=pos;i<=n;i+=i&(-i))
    ans+=c[i];
    return ans;
}
int main()
{
    in(n);in(m);
    for(int i=1;i<=n;++i)
    in(a[i]);
    for(int i=1;i<=m;++i)
    {
        int x,y,s,k;
        in(s);in(x);
        if(s==1)
        {
            in(y);in(k);
            addhead(y,k);
            addhead(x-1,-k);
        }
        else
        {
            printf("%d\n",sum(x));
        }
    }
}