二维树状数组 模板题 POJ1195

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

int s[1500][1500],n,m;
int lb(int i)
{
    return i&-i;
}

void set(int i,int j,int x)
{
    for(int ii=i;ii<=n;ii+=lb(ii))
    {
        for(int jj=j;jj<=m;jj+=lb(jj))
        {
            s[ii][jj]+=x;
        }
    }
}

int get(int i,int j)
{
    int ans=0;
    for(int ii=i;ii>0;ii-=lb(ii))
    {
        for(int jj=j;jj>0;jj-=lb(jj))
        {
            ans+=s[ii][jj];
        }
    }
    return ans;
}

int main()
{
    int o;
    while(scanf("%d",&o))
    {
        if(o==0)
        {
            scanf("%d",&n);
            m=n;
            memset(s,0,sizeof(s));
        }
        else if(o==1)
        {
            int x,y,a;
            scanf("%d%d%d",&x,&y,&a);
            set(x+1,y+1,a);
        }
        else if(o==2)
        {
            int l,b,r,t;
            scanf("%d%d%d%d",&l,&b,&r,&t);
            cout<<get(r+1,t+1)-get(r+1,b)-get(l,t+1)+get(l,b)<<endl;
        }
        else if(o==3)
            break;
    }
    return 0;
}