hdu 4348 To the moon(主席树区间操作)

 1 #include<bits/stdc++.h>
 2 #define F(i,a,b) for(int i=(a);i<=(b);++i)
 3 using namespace std;
 4 typedef long long ll;
 5 
 6 const int N=1e5+7;
 7 struct node{int l,r,lazy;ll sum;}T[N*40];
 8 int n,m,cnt,root[N],now;
 9 
10 void build(int &rt,int l=1,int r=n)
11 {
12     T[rt=++cnt].lazy=0;
13     if(l==r){scanf("%lld",&T[rt].sum);return;}
14     int m=l+r>>1;
15     build(T[rt].l,l,m),build(T[rt].r,m+1,r);
16     T[rt].sum=T[T[rt].l].sum+T[T[rt].r].sum;
17 }
18 
19 void update(int &x,int y,int L,int R,int v,int l=1,int r=n)
20 {
21     T[x=++cnt]=T[y];
22     if(L<=l&&r<=R){T[x].lazy+=v;return;}
23     int m=l+r>>1;
24     if(L<=m)update(T[x].l,T[y].l,L,R,v,l,m);
25     if(R>m)update(T[x].r,T[y].r,L,R,v,m+1,r);
26     int ll=max(L,l),rr=min(R,r);
27     T[x].sum+=1ll*(rr-ll+1)*v;
28 }
29 
30 ll query(int L,int R,int rt,ll add=0,int l=1,int r=n)
31 {
32     if(L<=l&&r<=R)return T[rt].sum+(T[rt].lazy+add)*(r-l+1);
33     int m=l+r>>1;ll an=0;
34     if(L<=m)an+=query(L,R,T[rt].l,add+T[rt].lazy,l,m);
35     if(R>m)an+=query(L,R,T[rt].r,add+T[rt].lazy,m+1,r);
36     return an;
37 }
38 
39 int main()
40 {
41     while(~scanf("%d%d",&n,&m))
42     {
43         cnt=0,now=0,build(root[0]);
44         F(i,1,m)
45         {
46             char op[2];
47             scanf("%s",op);
48             if(op[0]==‘C‘)
49             {
50                 int l,r,d;
51                 now++;
52                 scanf("%d%d%d",&l,&r,&d);
53                 update(root[now],root[now-1],l,r,d);
54             }
55             else if(op[0]==‘Q‘)
56             {
57                 int l,r;
58                 scanf("%d%d",&l,&r);
59                 printf("%lld\n",query(l,r,root[now]));
60             }
61             else if(op[0]==‘H‘)
62             {
63                 int l,r,t;
64                 scanf("%d%d%d",&l,&r,&t);
65                 printf("%lld\n",query(l,r,root[t]));
66             }else scanf("%d",&now),cnt=root[now+1];
67         }
68     }
69     return 0;
70 }