如何区间修改:
首先,我们不能进行线段树上的pushdown操作。因为这样做会让公共节点的值被修改。导致其他版本的线段树在查询时出现错误。
- 对于每一次区间修改操作,我们让落在L<l&&r<=R区间上的新开的节点的儿子为历史版本的节点,我们可以把这样的节点理解为分界点。因为修改区间将这段区间覆盖之后,它的孩子节点必然需要新开节点。
- 对新开的节点,除了孩子为旧节点的新开节点不需要pushup之外,其他的节点都需要pushup。
- 既然不能pushdown,那我们就在query函数中多加一个参数,让它带着lazy继续下传,用来模拟pushdown操作。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll maxn = 1e5+100;
ll n,m,a[maxn];
ll tot,rt[maxn];
struct node{ll ls,rs,sum,lazy;}t[maxn<<4];
inline void pushup(ll rt,ll len){
t[rt].sum = t[t[rt].ls].sum + t[t[rt].rs].sum +
t[t[rt].ls].lazy*(len-len/2) + t[t[rt].rs].lazy*(len/2);
}
ll build(ll l,ll r){
ll now = ++tot;
t[now].lazy=0;
if(l==r){
t[now].sum = a[l];
return now;
}
ll mid = l+r>>1;
t[now].ls = build(l,mid);
t[now].rs = build(mid+1,r);
t[now].sum = t[t[now].ls].sum + t[t[now].rs].sum;
return now;
}
ll update(ll rt,ll l,ll r,ll L,ll R,ll k){
ll now = ++tot;
t[now] = t[rt];
if(L<=l&&r<=R){
t[now].lazy+=k;
return now;
}
ll mid = l+r>>1;
if(L<=mid) t[now].ls = update(t[rt].ls,l,mid,L,R,k);
if(mid<R) t[now].rs = update(t[rt].rs,mid+1,r,L,R,k);
pushup(now,r-l+1);
return now;
}
ll query(ll rt,ll l,ll r,ll L,ll R,ll laz){
if(L<=l&&r<=R){
//因为不能pushdown,所以某个节点的值为sum+lazy*len
return t[rt].sum + (t[rt].lazy+laz)*(r-l+1);
}
ll mid = l+r>>1,ans=0;
if(L<=mid) ans+= query(t[rt].ls,l,mid,L,R,laz+t[rt].lazy);
if(mid<R) ans+= query(t[rt].rs,mid+1,r,L,R,laz+t[rt].lazy);
//lazy不能下传,我们让参数带着lazy下传
return ans;
}
int main(){
int flag=0;
while(scanf("%lld%lld",&n,&m)!=EOF){
if(flag) printf("\n");
++flag;
tot = 0;
ll now = 0;
for(ll i=1;i<=n;i++)
scanf("%lld",&a[i]);
rt[0] = build(1,n);
while(m--){
char op[10];
ll l,r,t;
scanf("%s",op);
if(op[0]=='Q'){
scanf("%lld%lld",&l,&r);
printf("%lld\n",query(rt[now],1,n,l,r,0));
}else if(op[0]=='C'){
scanf("%lld%lld%lld",&l,&r,&t);
now++;
rt[now] = update(rt[now-1],1,n,l,r,t);
}else if(op[0]=='H'){
scanf("%lld%lld%lld",&l,&r,&t);
printf("%lld\n",query(rt[t],1,n,l,r,0));
}else{
scanf("%lld",&now);
tot = rt[now+1]; //节省空间
}
}
}
return 0;
}