LibreOJ--135--二维树状数组--平面修改+平面查询

这是一道模板题。

给定一个大小为 N×MN×M 的零矩阵，直到输入文件结束，你需要进行若干个操作，操作有两类：

• 1 a b c d x，表示将左上角为 (a,b)(a,b)，右下角为 (c,d)(c,d) 的子矩阵全部加上 xx；

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

Input

第一行两个正整数 ，其中 n,mn,m 分别表示矩阵的行数与列数。

接下来若干行直到文件结束，均代表你需要进行的操作。

Output

对于每个 2 操作，输出一行代表查询的结果。

样例输入
4 4
1 1 1 3 3 2
1 2 2 4 4 1
2 2 2 3 3
样例输出
12

思路：参考：https://www.cnblogs.com/RabbitHu/p/BIT.html二维数组平面修改+平面查询；

1.申请四个树状数组空间；

2.更新：

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)){ 
			C1[i][j]+=k;
			C2[i][j]+=k*a;
			C3[i][j]+=k*b;
			C4[i][j]+=k*a*b;
	} 
}

3.求和：

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+=(x+1)*(y+1)*C1[i][j]-(y+1)*C2[i][j]-(x+1)*C3[i][j]+C4[i][j];
	return ans;
}

完整代码：

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<cstdio>
#define LL long long 
using namespace std;
const int maxa=1024*2+10;//~~pow(2,11)+10
LL n,m;
LL C1[maxa][maxa],C2[maxa][maxa],C3[maxa][maxa],C4[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)){ 
			C1[i][j]+=k;
			C2[i][j]+=k*a;
			C3[i][j]+=k*b;
			C4[i][j]+=k*a*b;
	} 
}
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+=(x+1)*(y+1)*C1[i][j]-(y+1)*C2[i][j]-(x+1)*C3[i][j]+C4[i][j];
	return ans;
}
int main(){
	scanf("%lld%lld",&n,&m);
	LL op,a,b,c,d,k; 
	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%lld%lld",&a,&b,&c,&d);
			printf("%lld\n",sum(c,d)-sum(a-1,d)-sum(c,b-1)+sum(a-1,b-1));//对于区间更新的查询 
		}
	}
	return 0;
}