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;
}