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

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

热度:74   发布时间:2023-12-12 06:12:04.0

这是一道模板题。

给出一个 n×m的零矩阵 AA,你需要完成如下操作:

  • 1 x y k:表示元素 Ax,y自增 k;
  • 2 a b c d:表示询问左上角为 (a,b),右下角为 (c,d) 的子矩阵内所有数的和。

Input

输入的第一行有两个正整数 n,m;
接下来若干行,每行一个操作,直到文件结束。

Output

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

样例输入
2 2
1 1 1 3
1 2 2 4
2 1 1 2 2
样例输出
7
Hint

思路:二维树状数组--单点修改+区间查询,只需改进一维树状数组即可;

参考:POJ1195:https://blog.csdn.net/queque_heiya/article/details/105895060

注意:LibreOJ的题目输出量大,尽量采用read()以及能用long不用int;

#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;//4096
LL a[maxa][maxa];
LL n,m;
LL lowbit(LL i){return i&-i;
}
void add(LL x,LL y,LL p){for(LL i=x;i<=n;i+=lowbit(i))for(LL j=y;j<=m;j+=lowbit(j))a[i][j]+=p;	
}
LL sum(int x,int y){LL ans=0;for(LL i=x;i>0;i-=lowbit(i))for(LL j=y;j>0;j-=lowbit(j))ans+=a[i][j];return ans;
} 
int main(){memset(a,0,sizeof a);scanf("%lld%lld",&n,&m);LL op,x,y,k,r,l;while(scanf("%lld",&op)!=EOF){if(op==1){scanf("%lld%lld%lld",&x,&y,&k);add(x,y,k);}if(op==2){scanf("%lld%lld%lld%lld",&x,&y,&r,&l);printf("%lld\n",sum(r,l)-sum(r,y-1)-sum(x-1,l)+sum(x-1,y-1));}}
}