树状数组

hdx
来个复杂点的,二维区间修改区间查询BIT

#include<cstdio>
#include<cctype>
#define maxn 2050
#define lowbit(a) a&(-a)
int n,m,e,f,g,h,delta;
char alpha;

void get(int &num){
	num=0;
	char ch;bool flag=0;
	for(;ch=getchar(),!isdigit(ch);)
		if(ch=='-') flag=1;
	for(num=ch-'0';ch=getchar(),isdigit(ch);num=num*10+ch-'0');
	(flag)&&(num=-num);
}

struct tree{
	int t[maxn][maxn];
	void add(int x,int y,int v){
		for(int i=x;i<=n;i+=lowbit(i))
			for(int j=y;j<=m;j+=lowbit(j))
				t[i][j]+=v;
	}

	int get(int x,int y){
		int ans=0;
		for(int i=x;i;i-=lowbit(i))
			for(int j=y;j;j-=lowbit(j))
				ans+=t[i][j];
		return ans;
	}
}a,b,c,d;

void add(int x1,int y1,int x2,int y2,int v){
	x2++,y2++;
	a.add(x1,y1,v);a.add(x1,y2,-v);a.add(x2,y1,-v);a.add(x2,y2,v);
	b.add(x1,y1,v*x1);b.add(x1,y2,-v*x1);b.add(x2,y1,-v*x2);b.add(x2,y2,v*x2);
	c.add(x1,y1,v*y1);c.add(x1,y2,-v*y2);c.add(x2,y1,-v*y1);c.add(x2,y2,v*y2);
	d.add(x1,y1,v*x1*y1);d.add(x1,y2,-v*x1*y2);d.add(x2,y1,-v*x2*y1);d.add(x2,y2,v*x2*y2);
}

int getnum(int x,int y){
	return (x+1)*(y+1)*a.get(x,y)-(y+1)*b.get(x,y)-(x+1)*c.get(x,y)+d.get(x,y);
}

int getans(int x1,int y1,int x2,int y2){
	x1--,y1--;
	return getnum(x2,y2)-getnum(x1,y2)-getnum(x2,y1)+getnum(x1,y1);
}

int main(){
	get(n),get(m);
	while(~scanf("\n%c",&alpha)){
		if(alpha=='L'){
			get(e),get(f),get(g),get(h),get(delta);
			add(e,f,g,h,delta);
		}
		else{
			get(e),get(f),get(g),get(h);
			printf("%d\n",getans(e,f,g,h));
		}
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值