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

Posted 2021-07-24 SSL_LKJ

区间修改区间查询

解题思路

二维差分+区间查询

AC代码

#include<cstdio>
using namespace std;
int n,m;
long long c[4][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[0][i][j]+=k;
	 	c[1][i][j]+=x*k;
	 	c[2][i][j]+=y*k;
	 	c[3][i][j]+=x*y*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[0][i][j]*(x+1)*(y+1)-c[1][i][j]*(y+1)-c[2][i][j]*(x+1)+c[3][i][j];
	return ans;
}
int main()
{
	int x,x1,y1,x2,y2,k;
	scanf("%d%d",&n,&m);
	while(scanf("%d",&x)!=EOF)
	 if(x==1)
	 {
		scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&k);
		update(x1,y1,1ll*k);//差分
	 	update(x1,y2+1,-1ll*k);
	 	update(x2+1,y1,-1ll*k);
	 	update(x2+1,y2+1,1ll*k);
	 }
	 else 
	 {
	 	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;	
} 

谢谢