bzoj2631: tree lct

  1 #include<iostream>
  2 #include<cstring>
  3 #include<cstdlib>
  4 #include<algorithm>
  5 #include<cstdio>
  6 
  7 using namespace std;
  8 
  9 const int D=15000000;
 10 char in[D],out[100010*10],*I=in,*O=out;
 11 #define gc (*I++)
 12 #define pc(x) ((*O++)=x)
 13 template <typename Q> 
 14 void gt(Q&x) {
 15     static char c,f;
 16     for(c=gc,f=0;!isdigit(c);c=gc)if(c==‘-‘) f=1;
 17     for(x=0;isdigit(c);c=gc) x=x*10+c-‘0‘;
 18     f && (x=-x);
 19 }
 20 template <typename Q>
 21 void pt(Q x){
 22     static char stk[20];
 23     static int top;
 24     top=0;
 25     if(x==0) pc(‘0‘);
 26     for(;x;x/=10) stk[++top] = x%10+‘0‘;
 27     for(;top;top--) pc(stk[top]);
 28 }
 29 
 30 typedef unsigned ll;
 31 const int Maxn=100010,Mod=51061;
 32 int n,m;
 33 ll w[Maxn],tagm[Maxn],taga[Maxn],sz[Maxn];
 34 int ch[Maxn][2],p[Maxn],flip[Maxn],sum[Maxn];
 35 
 36 void update(int x){
 37     if(x==0) return;
 38     int&l=ch[x][0],&r=ch[x][1];
 39     sum[x] = (sum[l] + sum[r] + w[x])%Mod;
 40     sz[x] = (sz[l] + sz[r] + 1)%Mod;
 41 }
 42 
 43 void add_tag(int x,ll a,ll m){
 44     if(!x) return;
 45     sum[x] = (sum[x]*m+a*sz[x])%Mod;
 46     w[x] = (w[x]*m+a)%Mod;
 47     tagm[x] = tagm[x]*m%Mod;
 48     taga[x] = (taga[x]*m+a)%Mod;
 49 }
 50 
 51 /*void tag_mul(int x,ll _v) {
 52     (sum[x]*=_v)%=Mod;
 53     (w[x]*=_v)%=Mod;
 54     (tagm[x]*=_v)%=Mod;
 55     (taga[x]*=_v)%=Mod;
 56 }
 57 
 58 void tag_add(int x,ll _v) {
 59     (sum[x]+=(sz[x]*_v)%Mod)%=Mod;
 60     (w[x]+=_v)%=Mod;
 61     (taga[x]+=_v)%=Mod;
 62 }*/
 63 
 64 bool isroot(int x) {
 65     return ch[p[x]][0]!=x && ch[p[x]][1]!=x;
 66 }
 67 
 68 void down(int x) {
 69     int &l=ch[x][0],&r=ch[x][1];
 70     if(flip[x]) {
 71         swap(l,r);
 72         flip[l]^=1;
 73         flip[r]^=1;
 74         flip[x]=0;
 75     }
 76     /*if(tagm[x]!=1) {
 77         for(int i=0;i<2;i++)if(ch[x][i])tag_mul(ch[x][i],tagm[x]);
 78         tagm[x]=1;
 79     }
 80     if(taga[x]!=0) {
 81         for(int i=0;i<2;i++)if(ch[x][i])tag_add(ch[x][i],taga[x]);
 82         taga[x]=0;
 83     }*/
 84     ll&a=taga[x],&m=tagm[x];
 85     if(a!=0 || m!=1) {
 86         add_tag(l,a,m);
 87         add_tag(r,a,m);
 88         a=0;m=1;
 89     }
 90 }
 91 
 92 void rotate(int x){
 93     int y=p[x],z=p[y];
 94     int l=ch[y][1]==x,r=l^1;
 95     if(!isroot(y)){
 96         ch[z][ch[z][1]==y]=x;
 97     }
 98     p[y]=x;
 99     p[ch[x][r]]=y;
100     p[x]=z;
101     
102     ch[y][l]=ch[x][r];
103     ch[x][r]=y;
104     
105     update(y);
106 //    update(x);
107 }
108 
109 int stk[Maxn],top;
110 void splay(int x){
111     stk[top=1]=x;
112     for(int t=x;!isroot(t);stk[++top]=t=p[t]);
113     for(;top;top--) down(stk[top]);
114     for(;!isroot(x);){
115         int y=p[x],z=p[y];
116         if(!isroot(y)) {
117             if( (ch[y][0]==x) ^ (ch[z][0]==y)) rotate(x);
118             else rotate(y);
119         }
120         rotate(x);
121     }
122     update(x);
123 }
124 
125 void access(int x) {
126     for(int t=0;x;x=p[t=x]){
127         splay(x);
128         ch[x][1]=t;
129         update(x);
130     }
131 }
132 
133 void newroot(int x) {
134     access(x);
135     splay(x);
136     flip[x]^=1;
137 }
138 
139 inline void n_as(int u,int v){
140     newroot(u);
141     access(v);
142     splay(v);
143 }
144 
145 void Cut(int x,int y) {
146     n_as(x,y);
147     ch[y][0]=p[x]=0;
148     update(x);
149 }
150 
151 void Link(int x,int y) {
152     newroot(x);
153     p[x]=y;
154 }
155 
156 int en[Maxn*2],next[Maxn*2],fir[Maxn];
157 void Add(int from,int to) {
158     static int tot=0;
159     en[++tot]=to;
160     next[tot]=fir[from];
161     fir[from]=tot;
162 }
163 
164 void BFS(int u) {
165     int *q=stk,ql=0,qr=0;
166     q[++qr]=u;
167     for(int x;ql<qr;){
168         x=q[++ql];
169         for(int k=fir[x];k;k=next[k]){
170             int v=en[k];
171             if(v==p[x]) continue;
172             p[v]=x;
173             q[++qr]=v;
174         }
175     }
176 }
177         
178 void init(){
179     gt(n),gt(m);
180     for(int u,v,i=1;i<n;i++) {
181         gt(u),gt(v);
182 //        Add(u,v),Add(v,u);
183         Link(u,v);
184     }
185 //    BFS(1);
186     tagm[0]=1;
187     for(int i=1;i<=n;i++) sz[i]=sum[i]=tagm[i]=w[i]=1;
188 }
189 
190 void work() {
191     char c;
192     ll u,v,d;
193     /*printf("round%d:\n",0);
194         for(int i=1;i<=n;i++) {
195             printf("%d :p=%d,ch=(%d,%d),w=%d,taga=%d,tagm=%d\n",i,p[i],ch[i][0],ch[i][1],w[i],taga[i],tagm[i]);
196         }*/
197     for(int i=1;i<=m;i++) {
198         for(;;) {
199             c=gc;
200             if(c==‘+‘) {
201                 gt(u),gt(v),gt(d);
202                 n_as(u,v);
203 //                tag_add(v,d);
204                 add_tag(v,d,1);
205                 break;
206             }if(c==‘-‘) {
207                 gt(u),gt(v);
208                 Cut(u,v);
209                 gt(u),gt(v);
210                 Link(u,v);
211                 break;
212             }if(c==‘*‘) {
213                 gt(u),gt(v),gt(d);
214                 n_as(u,v);
215                 add_tag(v,0,d);
216 //                tag_mul(v,d);
217                 break;
218             }if(c==‘/‘){
219                 gt(u),gt(v);
220                 n_as(u,v);
221                 pt(sum[v]);
222                 pc(‘\n‘);
223                 break;
224             }
225         }
226         /*printf("round%d:\n",i);
227         for(int i=1;i<=n;i++) {
228             printf("%d :p=%d,ch=(%d,%d),w=%d,taga=%d,tagm=%d,sum=%d\n",i,p[i],ch[i][0],ch[i][1],w[i],taga[i],tagm[i],sum[i]);
229         }*/
230     }
231 }
232 
233 int main() {
234 #ifdef DEBUG
235     freopen("in.txt","r",stdin);
236 //    freopen("out.txt","w",stdout);
237 #endif
238     fread(in,1,D,stdin);
239     
240     init();
241     work();
242     
243     return printf(out),0;
244 }