单点修改区间查询（树状数组）

Posted 2021-07-24 SSL_LKJ

单点修改区间查询

解题思路

就是典型的二维树状数组
用二维前缀和求答案

AC代码

#include<cstdio>
using namespace std;
int n,m;
long long c[5005][5005];
int lowbit(int x)
{
	return x&(-x);
}
void update(int x,int y,long long k)//存储
{
	for(int i=x;i<=n;i+=lowbit(i))
	 for(int j=y;j<=m;j+=lowbit(j))
	  c[i][j]+=k;
}
long long query(int x,int y)//查询
{
	long long ans=0;
	for(int i=x;i;i-=lowbit(i))
	 for(int j=y;j;j-=lowbit(j))
	  ans+=c[i][j];
	return ans;
}
int main()
{
	int x,y,k;
	scanf("%d%d",&n,&m);
	while(scanf("%d",&x)!=EOF)
	 if(x==1)
	 {
		scanf("%d%d%d",&x,&y,&k);
		update(x,y,1ll*k);
	 }
	 else 
	 {
	 	int x1,y1,x2,y2;
	 	scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
		printf("%lld\\n",query(x2,y2)-query(x1-1,y2)-query(x2,y1-1)+query(x1-1,y1-1));//二维前缀和
	 }
	return 0;	
} 

谢谢