树状数组模板

一维树状数组：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,m;
int a[500005];

int lowbit(int x)
{
    return x&(-x);
}

void update(int x,int k)
{
    while(x<=n)
    {
        a[x]+=k;
        x+=lowbit(x);
    }
}

int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=a[x];
        x-=lowbit(x);
    }
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        update(i,x);
    }
    for(int i=1;i<=m;i++)
    {
        int z,x,y;
        scanf("%d%d%d",&z,&x,&y);
        if(z==1) update(x,y);
        else printf("%d
",sum(y)-sum(x-1));
    }
    return 0;
}

 树状数组拓展：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,m;
int a[500005];

int lowbit(int x)
{
    return x&(-x);
}

void update(int x,int k)
{
    while(x<=n)
    {
        a[x]+=k;
        x+=lowbit(x);
    }
}

int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=a[x];
        x-=lowbit(x);
    }
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    int last=0,now;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&now);
        update(i,now-last);
        last=now;
    }
    for(int i=1;i<=m;i++)
    {
        int z,x,y,k;
        scanf("%d",&z);
        if(z==1)
        {
            scanf("%d%d%d",&x,&y,&k);
            update(x,k);
            update(y+1,-k);
        }
        else
        {
            scanf("%d",&x);
            printf("%d
",sum(x));
        }
    }
    return 0;
}