简介:

一维的树状数组已经说过了,下面简单记录一下二维树状数组的使用方式,以便复习

关键函数

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

void update(int x,int y,ll v){
    for(int i = x ; i <= n ; i += lowbit(i)){
        for(int j = y ; j <= n ; j += lowbit(j)){
            c[i][j] += v;
        }
    }
}

int getsum(int x,int y){
    int ans = 0;
    for(int i = x ; i > 0 ; i -= lowbit(i)){
        for(int j = y ; j > 0 ; j -= lowbit(j)){
            ans += c[i][j];
        }
    }
    return ans;
}

注意:

二维树状数组求区间和时,要注意直接减去左侧区间的和,这个画个坐标系就知道了

模板题

打鼹鼠(mole)

Description

SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊……),于是便无聊地开始玩“打鼹鼠”……

在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。

Input

第一行,一个数n,表示鼹鼠的范围。

以后每一行开头都有一个数m,表示不同的操作:
m=1,那么后面跟着3个数x,y,k(0<=x,y<n),表示在点(x,y)处新出现了k只鼹鼠;
m=2,那么后面跟着4个数x1,y1,x2,y2(0<=x1<=x2<n,0<=y1<=y2<n),表示询问矩形(x1,y1)-(x2,y2)内的鼹鼠数量;
m=3,表示老师来了,不能玩了。保证这个数会在输入的最后一行。

询问数不会超过10000,鼹鼠数不会超过max long int。

Output

对于每个m=2,输出一行数,这行数只有一个数,即所询问的区域内鼹鼠的个数。

Sample Input 1

4
1 2 2 5
2 0 0 2 3
3

Sample Output 1

5

代码:

#include<bits/stdc++.h>

using namespace std;
int n,m;

const int maxn = 1030;
typedef long long ll; 
ll c[maxn][maxn];

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

void update(int x,int y,ll v){
    for(int i = x ; i <= n ; i += lowbit(i)){
        for(int j = y ; j <= n ; j += lowbit(j)){
            c[i][j] += v;
        }
    }
}

ll getsum(int x,int y){
    ll ans = 0;
    for(int i = x ; i > 0 ; i -= lowbit(i)){
        for(int j = y ; j > 0 ; j -= lowbit(j)){
            ans += c[i][j];
        }
    }
    return ans;
}

int main(){
    cin >> n;
    while(true){
        int m;
        cin >> m;
        if(m == 1){
            int x,y;
            ll k;
            cin >> x >> y >> k;
            update(x+1,y+1,k);
        }else if(m == 2){
            int x,y,xx,yy;
            cin >> x >> y >> xx >> yy;
            x++,y++,xx++,yy++;
            cout <<getsum(xx,yy) - getsum(x-1,yy) - getsum(xx,y-1) + getsum(x-1,y-1) << endl;
        }else if(m == 3) return 0;
    }
    return 0;
}