LibreOJ--134--二维树状数组--平面修改+单点查询

这是一道模板题。

给出一个 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
#include
#include
#include
#include
#include
#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));//对于区间更新的查询 
		}
	}
}

 

你可能感兴趣的:(#,LibreOJ模板题,#,常用模板代码)