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

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

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define lson v[x].ls,L,mid
 4 #define rson v[x].rs,mid+1,R
 5 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 6 typedef long long ll;
 7 using namespace std;
 8 
 9 const int N=100010;
10 char op;
11 int n,m,l,r,k,tim,nd,a[N],rt[N];
12 struct Tr{ int ls,rs; ll sm,tag; }v[N*25];
13 
14 void put(int &x,int L,int R,ll k){ if (x) v[++nd]=v[x],x=nd,v[nd].sm+=(R-L+1)*k,v[nd].tag+=k; }
15 
16 void push(int x,int L,int R){ int mid=(L+R)>>1; if (v[x].tag) put(lson,v[x].tag),put(rson,v[x].tag),v[x].tag=0; }
17 
18 void build(int &x,int L,int R){
19     x=++nd;
20     if (L==R){ v[x]=(Tr){0,0,a[L],0}; return; }
21     int mid=(L+R)>>1;
22     build(lson); build(rson);
23     v[x].sm=v[v[x].ls].sm+v[v[x].rs].sm; v[x].tag=0;
24 }
25 
26 void ins(int y,int &x,int L,int R,int l,int r,int k){
27     x=++nd; v[x]=v[y];
28     if (L==l && r==R){ v[x].sm+=1ll*(R-L+1)*k; v[x].tag+=k; return; }
29     int mid=(L+R)>>1; push(x,L,R);
30     if (r<=mid) ins(v[y].ls,lson,l,r,k);
31     else if (l>mid) ins(v[y].rs,rson,l,r,k);
32         else ins(v[y].ls,lson,l,mid,k),ins(v[y].rs,rson,mid+1,r,k);
33     v[x].sm=v[v[x].ls].sm+v[v[x].rs].sm;
34 }
35 
36 ll que(int x,int L,int R,int l,int r){
37     if (L==l && r==R) return v[x].sm;
38     int mid=(L+R)>>1; push(x,L,R);
39     if (r<=mid) return que(lson,l,r);
40     else if (l>mid) return que(rson,l,r);
41         else return que(lson,l,mid)+que(rson,mid+1,r);
42 }
43 
44 int main(){
45     freopen("hdu4348.in","r",stdin);
46     freopen("hdu4348.out","w",stdout);
47     while (~scanf("%d%d",&n,&m)){
48         rep(i,1,n) scanf("%d",&a[i]);
49         nd=tim=0; build(rt[0],1,n);
50         rep(i,1,m){
51             scanf(" %c",&op);
52             if (op=='C') scanf("%d%d%d",&l,&r,&k),tim++,ins(rt[tim-1],rt[tim],1,n,l,r,k);
53             if (op=='Q') scanf("%d%d",&l,&r),printf("%lld\n",que(rt[tim],1,n,l,r));
54             if (op=='H') scanf("%d%d%d",&l,&r,&k),printf("%lld\n",que(rt[k],1,n,l,r));
55             if (op=='B') scanf("%d",&k),tim=k;
56         }
57         puts("");
58     }
59     return 0;
60 }
未永久化(MLE)

相关文章: