学可持久化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 }