题目链接:https://vjudge.net/problem/POJ-3468
题意:区间修改和查询。
思路:区间修改时一定要下传标记,这样才能更新上层节点的值,否则会出错。
如 5 3
1 1 1 1 1
C 1 2 1
C 2 3 1
Q 1 2
如果在区间修改时不下传标记,当对执行C 2 3 1时,此时4号节点也就是1和2的父节点的lazy没有下传导致pushup操作时4号节点的sum会出错。
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
#include <iostream>
#include <cmath>
#include <set>
#include <fstream>
#include <list>
#include <stack>
#include <map>
using namespace std;
const long long maxn=1e5+5;
long long p[maxn];
struct NODE
{
long long l,r;
long long lazy,sum;
NODE()
{
l=r=lazy=sum=0;
}
} a[maxn*4];
void pushup(long long k)
{
a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void build(long long l,long long r,long long k)
{
a[k].l=l;
a[k].r=r;
if(l==r)
{
a[k].sum=p[l];
return ;
}
long long mid=(l+r)/2;
build(l,mid,k*2);
build(mid+1,r,k*2+1);
pushup(k);
}
void pushdown(long long k)
{
if(a[k].l==a[k].r)
{
a[k].lazy=0;
return ;
}
a[k*2].sum+=(a[k*2].r-a[k*2].l+1)*a[k].lazy;
a[k*2+1].sum+=(a[k*2+1].r-a[k*2+1].l+1)*a[k].lazy;
a[k*2].lazy+=a[k].lazy;
a[k*2+1].lazy+=a[k].lazy;
a[k].lazy=0;
}
void add(long long l,long long r,long long L,long long R,long long k,long long val)
{
if(l>R||r<L)
return ;
if(l>=L&&r<=R)
{
a[k].sum+=(a[k].r-a[k].l+1)*val;
a[k].lazy+=val;
return ;
}
long long mid=(l+r)/2;
pushdown(k);
add(l,mid,L,R,k*2,val);
add(mid+1,r,L,R,k*2+1,val);
pushup(k);
}
long long query(long long l,long long r,long long L,long long R,long long k)
{
if(l>R||r<L)
return 0;
pushdown(k);
if(l>=L&&r<=R)
return a[k].sum;
long long mid=(l+r)/2;
long long cnt=0;
cnt+=query(l,mid,L,R,k*2);
cnt+=query(mid+1,r,L,R,k*2+1);
return cnt;
}
int main()
{
memset(a,0,sizeof(a));
long long n,q;
scanf("%lld%lld",&n,&q);
for(long long i=1; i<=n; i++)
{
scanf("%lld",&p[i]);
}
build(1,n,1);
char op[10];
long long x,y;
long long z;
while(q--)
{
scanf("%s",op);
if(op[0]=='Q')
{
scanf("%lld%lld",&x,&y);
printf("%lld\n",query(1,n,x,y,1));
}
else
{
scanf("%lld%lld%lld",&x,&y,&z);
add(1,n,x,y,1,z);
}
}
return 0;
}