1540：打鼹鼠 - 二维树状数组（单点更新+区间查询）

1540：打鼹鼠_二维树状数组

时间限制: 1000 ms         内存限制: 524288 KB
提交数: 185     通过数: 37

【题目描述】

这是一道模板题。

给出一个 n×m

的零矩阵 A，你需要完成如下操作：

1 x y k：表示元素 Ax,y自增 k；

2 a b c d：表示询问左上角为 (a,b)，右下角为 (c,d) 的子矩阵内所有数的和。

【输入】

输入的第一行有两个正整数 n,m；

接下来若干行，每行一个操作，直到文件结束。

【输出】

对于每个 2

操作，输出一个整数，表示对于这个操作的回答。

【输入样例】

2 2
1 1 1 3
1 2 2 4
2 1 1 2 2

【输出样例】

7

【提示】

数据范围与提示：

对于 10% 的数据，n=1；

对于另 10% 的数据，m=1；

对于全部数据，1≤n,m≤212,1≤x,a,c≤n,1≤y,b,d≤m,∣k∣≤105，保证操作数目不超过 3×105 ，且询问的子矩阵存在。

思路：

二维树状数组的“单点更新+区间查询板子题”

学习博客：这个大佬讲的特别好

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=5000;
ll c[N+5][N+5];

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

void add(int x,int y,int z){
    int y1;
    while(x<=N){
        y1=y;
        while(y1<=N){
            c[x][y1]+=z;
            y1+=lowbit(y1);
        }
        x+=lowbit(x);
    }
}

ll sum(int x,int y){
    ll res=0,y1=y;
    while(x>0){
        y1=y;
        while(y1>0){
            res+=c[x][y1];
            y1-=lowbit(y1);
        }
        x-=lowbit(x);
    }
    return res;
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    int k;
    while(scanf("%d",&k)!=EOF){
        if(k==1){
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
        }
        else{
            int a,b,c,d;
            scanf("%d%d%d%d",&a,&b,&c,&d);
            ll ans=sum(c,d)-sum(a-1,d)-sum(c,b-1)+sum(a-1,b-1);
            printf("%lld\n",ans);
        }
    }
}