P3372 模板线段树 1（区间修改区间查询）（树状数组）

Posted 2021-07-24 SSL_LKJ

P3372 【模板】线段树 1（区间修改区间查询）

题目传送门

解题思路

这题除了用线段树做
还可以用树状数组做
主要就是利用二维树状数组进行维护

AC代码

#include<cstdio>
using namespace std;
int n,T;
long long c[3][1000005];
int lowbit(int x)
{
	return x&(-x);
}
void update(int x,long long y)//存储
{
	int xx=x;
	for(;x<=n;x+=lowbit(x))c[1][x]+=y,c[2][x]+=y*xx;
}
long long query(int x)//查询
{
	long long sum=0;
	int xx=x;
	for(;x;x-=lowbit(x))sum+=c[1][x]*(xx+1ll)-c[2][x];
	return sum;
}
int main()
{
	scanf("%d%d",&n,&T);
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		update(i,1ll*x);
		update(i+1,-1ll*x);
	}
	while(T--)
	{
		int t,l,r,x;
		scanf("%d",&t);
		if(t==1)
		{
			scanf("%d%d%d",&l,&r,&x);
			update(l,1ll*x);
			update(r+1,-1ll*x);
		}
		else 
		{
			scanf("%d%d",&l,&r);
			printf("%lld\\n",query(r)-query(l-1));//输出
		}
	}
	return 0;
}

谢谢