POJ 1195 二维树状数组

模板

#include <stdio.h>
#include <iostream>
#include <queue>
#include <algorithm>
#include <map>
#include <vector>
#include <cmath>
#include <string.h>
using namespace std;

#define ll long long

#define P pair<int,int>
#define PDI pair<double,int>
#define fst first
#define sec second
#define MS(x) memset(x,0,sizeof(x))
#define MAX 400000
#define INF 0x3f3f3f3f

int N;
struct bit
{
    ll data[1100][1100];
    void add(int x,int y,ll num)
    {
        for(int i=x;i<=N;i+=i&(-i))
            for(int j=y;j<=N;j+=j&(-j))
                data[i][j]+=num;//,cout<<data[x][y]<<endl;
    }
    ll sum(int x,int y)
    {
        ll res=0;
        for(int i=x;i>0;i-=i&(-i))
            for(int j=y;j>0;j-=j&(-j))
                res+=data[i][j];//,cout<<data[x][y]<<endl;
        return res;
    }
    
}area;

int main()
{
    int t;
    while(scanf("%d%d",&t,&N)!=EOF)
    {
        memset(area.data,0,sizeof(area.data));
        while(scanf("%d",&t)&&t!=3)
        {
            if(t==1)
            {
                int x,y,a;
                scanf("%d%d%d",&x,&y,&a);
                area.add(x+1,y+1,a);
            }else
            {
                int x1,y1,x2,y2;
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                x1++,x2++,y1++,y2++;
                ll ans=area.sum(x2,y2)-area.sum(x1-1,y2)-area.sum(x2,y1-1)+area.sum(x1-1,y1-1);
                printf("%lld\n",ans);
            }
        }
    }
    return 0;
}