[HDU4348]To the moon(主席树+标记永久化)

学可持久化treap的时候才发现自己竟然没写过需要标记下传的主席树，然而现在发现大部分操作都可以标记永久化，下传会增大占用空间。

这题一种写法是和普通的线段树一样标记下传，注意所有修改操作（包括put()）都要新建点。于是MLE了。

 #include<cstdio>
 #include<algorithm>
 #define lson v[x].ls,L,mid
 #define rson v[x].rs,mid+1,R
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 typedef long long ll;
 using namespace std;

 const int N=;
 char op;
 int n,m,l,r,k,tim,nd,a[N],rt[N];
 struct Tr{ int ls,rs; ll sm,tag; }v[N*];

 void put(int &x,int L,int R,ll k){ if (x) v[++nd]=v[x],x=nd,v[nd].sm+=(R-L+)*k,v[nd].tag+=k; }

 void push(int x,int L,int R){ int mid=(L+R)>>; if (v[x].tag) put(lson,v[x].tag),put(rson,v[x].tag),v[x].tag=; }

 void build(int &x,int L,int R){
     x=++nd;
     if (L==R){ v[x]=(Tr){,,a[L],}; return; }
     int mid=(L+R)>>;
     build(lson); build(rson);
     v[x].sm=v[v[x].ls].sm+v[v[x].rs].sm; v[x].tag=;
 }

 void ins(int y,int &x,int L,int R,int l,int r,int k){
     x=++nd; v[x]=v[y];
     if (L==l && r==R){ v[x].sm+=1ll*(R-L+)*k; v[x].tag+=k; return; }
     int mid=(L+R)>>; push(x,L,R);
     if (r<=mid) ins(v[y].ls,lson,l,r,k);
     else if (l>mid) ins(v[y].rs,rson,l,r,k);
         else ins(v[y].ls,lson,l,mid,k),ins(v[y].rs,rson,mid+,r,k);
     v[x].sm=v[v[x].ls].sm+v[v[x].rs].sm;
 }

 ll que(int x,int L,int R,int l,int r){
     if (L==l && r==R) return v[x].sm;
     int mid=(L+R)>>; push(x,L,R);
     if (r<=mid) return que(lson,l,r);
     else if (l>mid) return que(rson,l,r);
         else return que(lson,l,mid)+que(rson,mid+,r);
 }

 int main(){
     freopen("hdu4348.in","r",stdin);
     freopen("hdu4348.out","w",stdout);
     while (~scanf("%d%d",&n,&m)){
         rep(i,,n) scanf("%d",&a[i]);
         nd=tim=; build(rt[],,n);
         rep(i,,m){
             scanf(" %c",&op);
             if (op=='C') scanf("%d%d%d",&l,&r,&k),tim++,ins(rt[tim-],rt[tim],,n,l,r,k);
             if (op=='Q') scanf("%d%d",&l,&r),printf("%lld\n",que(rt[tim],,n,l,r));
             if (op=='H') scanf("%d%d%d",&l,&r,&k),printf("%lld\n",que(rt[k],,n,l,r));
             if (op=='B') scanf("%d",&k),tim=k;
         }
         puts("");
     }
     return ;
 }

未永久化(MLE)

另一种写法就是标记永久化，若一个修改区间覆盖当前区间则将tag+=k，但并不下传。询问时将答案加上tag的贡献即可。

注意多组数据的清空问题。

 #include<cstdio>
 #include<algorithm>
 #define lson v[x].ls,L,mid
 #define rson v[x].rs,mid+1,R
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 typedef long long ll;
 using namespace std;

 const int N=;
 char op;
 int n,m,l,r,k,tim,nd,flag,a[N],rt[N];
 struct Tr{ int ls,rs; ll sm,tag; }v[N*];

 void build(int &x,int L,int R){
     x=++nd;
     if (L==R){ v[x]=(Tr){,,a[L],}; return; }
     int mid=(L+R)>>;
     build(lson); build(rson);
     v[x].sm=v[v[x].ls].sm+v[v[x].rs].sm; v[x].tag=;
 }

 void ins(int y,int &x,int L,int R,int l,int r,int k){
     x=++nd; v[x]=v[y]; v[x].sm+=1ll*(r-l+)*k;
     if (L==l && r==R){ v[x].tag+=k; return; }
     int mid=(L+R)>>;
     if (r<=mid) ins(v[y].ls,lson,l,r,k);
     else if (l>mid) ins(v[y].rs,rson,l,r,k);
         else ins(v[y].ls,lson,l,mid,k),ins(v[y].rs,rson,mid+,r,k);
 }

 ll que(int x,int L,int R,int l,int r){
     if (L==l && r==R) return v[x].sm;
     int mid=(L+R)>>,res=v[x].tag*(r-l+);
     if (r<=mid) return res+que(lson,l,r);
     else if (l>mid) return res+que(rson,l,r);
         else return res+que(lson,l,mid)+que(rson,mid+,r);
 }

 int main(){
     freopen("hdu4348.in","r",stdin);
     freopen("hdu4348.out","w",stdout);
     while (~scanf("%d%d",&n,&m)){
         if (flag) puts(""); else flag=;
         rep(i,,n) scanf("%d",&a[i]);
         nd=tim=; build(rt[],,n);
         rep(i,,m){
             scanf(" %c",&op);
             if (op=='C') scanf("%d%d%d",&l,&r,&k),tim++,ins(rt[tim-],rt[tim],,n,l,r,k);
             if (op=='Q') scanf("%d%d",&l,&r),printf("%lld\n",que(rt[tim],,n,l,r));
             if (op=='H') scanf("%d%d%d",&l,&r,&k),printf("%lld\n",que(rt[k],,n,l,r));
             if (op=='B') scanf("%d",&tim);
         }
     }
     return ;
 }