这是一道模板题。
给出一个 n×mn×m 的零矩阵 AA,你需要完成如下操作:
1 a b c d k
:表示左上角为 (a,b)(a,b),右下角为 (c,d)(c,d) 的子矩阵内所有数都自增 kk;2 x y
:表示询问元素 Ax,yAx,y 的值;
Input
输入的第一行有两个正整数 n,mn,m;
接下来若干行,每行一个操作,直到文件结束。
Output
对于每个 2
操作,输出一个整数,表示对于这个操作的回答。
样例输入
2 2
1 1 1 2 2 5
1 1 2 2 2 -3
2 1 1
2 1 2
2 2 1
2 2 2
样例输出
5
2
5
2
思路:1.结合一维树状数组改写即可;2.求和直接sum(x,y);3.差分数组直接初始化为0即可;
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<cstdio>
#define LL long long
using namespace std;
const int maxa=1024*4+10;//~~pow(2,12)+10
LL n,m;
LL C[maxa][maxa];
LL lowbit(LL i){return i&-i;
}
void add(LL a,LL b,LL k){for(LL i=a;i<=n;i+=lowbit(i))for(LL j=b;j<=m;j+=lowbit(j))C[i][j]+=k;
}
LL sum(LL x,LL y){LL ans=0;for(LL i=x;i>0;i-=lowbit(i))for(LL j=y;j>0;j-=lowbit(j))ans+=C[i][j];return ans;
}
int main(){scanf("%lld%lld",&n,&m);LL op,a,b,c,d,k,x,y; memset(C,0,sizeof C);while(scanf("%lld",&op)!=EOF){if(op==1){scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&k);add(a,b,k);add(c+1,d+1,k);add(c+1,b,-k);add(a,d+1,-k);//一维差分数组应该为:add(x,k),add(y+1,-k); }else if(op==2){scanf("%lld%lld",&x,&y);printf("%lld\n",sum(x,y));//对于区间更新的查询 }}
}