比之前那道LCT裸题难了一点儿。
这道题除了维护树的形态和连通性，还要维护点权了。
所以记录size、sum什么的，跟splay差不多。
lazy_tag、pushup、pushdown什么的，跟线段树差不多。
就是一道大杂烩嘛......
各种乱搞。
这里不再写cut(x,y)函数了，而是写了isolate(x,y)函数。
意为把从x到y的链单独提取到一个splay里面。
再加个删除父子关系不就是cut了嘛。
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #define M 51061 5 #define ll long long 6 #define id(x) (s[f[x]][1]==x) 7 using namespace std; 8 9 int n,q; 10 int s[100005][2],f[100005],sz[100005]; 11 bool rev[100005],rt[100005]; 12 ll v[100005],sum[100005],lza[100005],lzm[100005]; 13 14 void reverse(int p) 15 { 16 if(!p)return; 17 swap(s[p][0],s[p][1]); 18 rev[p]^=1; 19 } 20 21 void pushup(int p) 22 { 23 sz[p]=sz[s[p][0]]+sz[s[p][1]]+1; 24 sum[p]=(sum[s[p][0]]+sum[s[p][1]]+v[p])%M; 25 } 26 27 void mul(int p,ll c) 28 { 29 sum[p]=sum[p]*c%M; 30 v[p]=v[p]*c%M; 31 lzm[p]=lzm[p]*c%M; 32 lza[p]=lza[p]*c%M; 33 } 34 35 void plus(int p,ll c) 36 { 37 sum[p]=(sum[p]+sz[p]*c%M)%M; 38 v[p]=(v[p]+c)%M; 39 lza[p]=(lza[p]+c)%M; 40 } 41 42 void pushdown(int p) 43 { 44 if(lzm[p]!=1)mul(s[p][0],lzm[p]),mul(s[p][1],lzm[p]),lzm[p]=1; 45 if(lza[p])plus(s[p][0],lza[p]),plus(s[p][1],lza[p]),lza[p]=0; 46 if(rev[p])reverse(s[p][0]),reverse(s[p][1]),rev[p]=0; 47 } 48 49 void rotate(int p) 50 { 51 int k=id(p); 52 int fa=f[p]; 53 if(rt[fa])rt[p]=1,rt[fa]=0; 54 else s[f[fa]][id(fa)]=p; 55 s[fa][k]=s[p][!k]; 56 s[p][!k]=fa; 57 f[p]=f[fa]; 58 f[fa]=p; 59 f[s[fa][k]]=fa; 60 pushup(fa); 61 pushup(p); 62 } 63 64 void down(int p) 65 { 66 if(!rt[p])down(f[p]); 67 pushdown(p); 68 } 69 70 void splay(int p) 71 { 72 down(p); 73 while(!rt[p]) 74 { 75 int fa=f[p]; 76 if(rt[fa]) 77 { 78 rotate(p); 79 return; 80 } 81 if(id(p)^id(fa))rotate(p); 82 else rotate(fa); 83 rotate(p); 84 } 85 } 86 87 void access(int p) 88 { 89 int son=0; 90 while(p) 91 { 92 splay(p); 93 rt[s[p][1]]=1,rt[son]=0; 94 s[p][1]=son; 95 pushup(p); 96 son=p,p=f[p]; 97 } 98 } 99 100 void mtr(int p) 101 { 102 access(p); 103 splay(p); 104 reverse(p); 105 } 106 107 void link(int x,int y) 108 { 109 mtr(x); 110 f[x]=y; 111 } 112 113 void isolate(int x,int y) 114 { 115 mtr(x); 116 access(y); 117 splay(y); 118 } 119 120 int main() 121 { 122 scanf("%d%d",&n,&q); 123 for(int i=1;i<=n;i++)lzm[i]=sz[i]=v[i]=rt[i]=1; 124 for(int i=1;i<n;i++) 125 { 126 int x,y; 127 scanf("%d%d",&x,&y); 128 link(x,y); 129 } 130 while(q--) 131 { 132 char op[5]; 133 scanf("%s",op+1); 134 int pu,pv,nu,nv; 135 ll c; 136 if(op[1]=='+') 137 { 138 scanf("%d%d%lld",&pu,&pv,&c); 139 isolate(pu,pv); 140 plus(pv,c); 141 } 142 if(op[1]=='-') 143 { 144 scanf("%d%d%d%d",&pu,&pv,&nu,&nv); 145 isolate(pu,pv); 146 s[pv][0]=f[pu]=0; 147 rt[pu]=1; 148 link(nu,nv); 149 } 150 if(op[1]=='*') 151 { 152 scanf("%d%d%lld",&pu,&pv,&c); 153 isolate(pu,pv); 154 mul(pv,c); 155 } 156 if(op[1]=='/') 157 { 158 scanf("%d%d",&pu,&pv); 159 isolate(pu,pv); 160 printf("%lld ",sum[pv]); 161 } 162 } 163 return 0; 164 }