Vijos1512 SuperBrother打鼹鼠

SuperBrother打鼹鼠

Vijos链接

题目描述：

在一个矩阵中，有三种操作：

1.后面跟着3个数x,y,k，表示在点(x,y)处新出现了k只鼹鼠。

2.后面跟着4个数x1,y1,x2,y2，表示询问矩形(x1,y1)-(x2,y2)内的鼹鼠数量。

3.表示结束。

思路：

一个树状数组就可以搞定了，不过需要二维的树状数组，才能实现二维的区间查询。

代码：

#include<cstdio>
long long n,m,dis[10010][10010],tot,a,b,c,d;
using namespace std;
void add(long long x,long long y,long long z){
    for(long long i=x;i<=n;i+=i&(-i))
        for(long long j=y;j<=n;j+=j&(-j))
            dis[i][j]+=z;
}
long long q(long long x,long long y){
    tot=0;
    for(long long i=x;i;i-=i&(-i))
        for(long long j=y;j;j-=j&(-j))
            tot+=dis[i][j];
    return tot;
}
int main(){
    scanf("%lld",&n);
    while(1){
        scanf("%lld",&m);
        if(m==1){
            scanf("%lld%lld%lld",&a,&b,&c);
            a++;b++;
            add(a,b,c);
        }
        else
        if(m==2){
            scanf("%lld%lld%lld%lld",&a,&b,&c,&d);
            a++;b++;c++;d++;
            printf("%lld
",q(c,d)-q(a-1,d)-q(c,b-1)+q(a-1,b-1));
        }
        else
            break;
    }
    return 0;
}