当前位置: 代码迷 >> 综合 >> LibreOJ--135--二维树状数组--平面修改+平面查询
  详细解决方案

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

热度:15   发布时间:2023-12-12 06:11:40.0

这是一道模板题。

给定一个大小为 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;
}