题目大意:给你一棵树,让你维护一个数据结构,支持
边的断,连
树链上所有点点权加上某个值
树链上所有点点权乘上某个值
求树链所有点点权和
(辣鸡bzoj又是土豪题,洛谷P1501传送门)
LCT裸题,下传标记,搞法类似于洛谷线段树模板2
先下传乘法标记,val,sum,乘法标记,加法标记都要乘
再下传加法标记,val,加法标记直接加,sum应该加上子树size*加的值
卡了10分钟我发现打标记应该同时更新这个位置的实际值
又卡了10分钟发现我下传完标记忘记清零了
然而我又忘记了一个事情被卡了20分钟!乘法标记能下传的条件是,乘法标记不等于1 而不是 不等于0!!!
智商已下线
1 #include <cstdio> 2 #include <algorithm> 3 #include <cstring> 4 #define il inline 5 #define ui unsigned int 6 #define mod 51061 7 #define N 101000 8 using namespace std; 9 10 int n,m,tp; 11 int stk[N]; 12 struct LinkCutTree{ 13 int fa[N],ch[N][2],rv[N]; 14 ui sz[N],sum[N],val[N],add[N],mul[N]; 15 il int idf(int x){return ch[fa[x]][0]==x?0:1;} 16 il void con(int x,int ff,int p){fa[x]=ff,ch[ff][p]=x;} 17 il int isroot(int x){return (ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x)?1:0;} 18 il void rev(int x){swap(ch[x][0],ch[x][1]),rv[x]^=1;} 19 il void pushup(int x) 20 { 21 sum[x]=(sum[ch[x][0]]+sum[ch[x][1]]+val[x])%mod; 22 sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1; 23 } 24 il void pushdown(int x) 25 { 26 #define ls ch[x][0] 27 #define rs ch[x][1] 28 if(mul[x]!=1) 29 { 30 sum[ls]=(sum[ls]*mul[x])%mod,sum[rs]=(sum[rs]*mul[x])%mod; 31 val[ls]=(val[ls]*mul[x])%mod,val[rs]=(val[rs]*mul[x])%mod; 32 mul[ls]=(mul[ls]*mul[x])%mod,mul[rs]=(mul[rs]*mul[x])%mod; 33 add[ls]=(add[ls]*mul[x])%mod,add[rs]=(add[rs]*mul[x])%mod; 34 mul[x]=1; 35 } 36 if(add[x]) 37 { 38 sum[ls]=(sum[ls]+(sz[ls]%mod*add[x])%mod)%mod; 39 sum[rs]=(sum[rs]+(sz[rs]%mod*add[x])%mod)%mod; 40 val[ls]=(val[ls]+add[x])%mod; 41 val[rs]=(val[rs]+add[x])%mod; 42 add[ls]=(add[ls]+add[x])%mod; 43 add[rs]=(add[rs]+add[x])%mod; 44 add[x]=0; 45 } 46 if(rv[x]) 47 { 48 if(ch[x][0]) rev(ch[x][0]); 49 if(ch[x][1]) rev(ch[x][1]); 50 rv[x]=0; 51 } 52 #undef ls 53 #undef rs 54 } 55 il void rot(int x) 56 { 57 int y=fa[x];int ff=fa[y];int px=idf(x);int py=idf(y); 58 if(!isroot(y)) ch[ff][py]=x; 59 fa[x]=ff;con(ch[x][px^1],y,px),con(y,x,px^1); 60 pushup(y),pushup(x); 61 } 62 void splay(int x) 63 { 64 int y=x;stk[++tp]=x; 65 while(!isroot(y)){stk[++tp]=fa[y],y=fa[y];} 66 while(tp){pushdown(stk[tp--]);} 67 while(!isroot(x)) 68 { 69 y=fa[x]; 70 if(isroot(y)) rot(x); 71 else if(idf(y)==idf(x)) rot(y),rot(x); 72 else rot(x),rot(x); 73 } 74 } 75 void access(int x){for(int y=0;x;y=x,x=fa[x])splay(x),ch[x][1]=y,pushup(x);} 76 //int findrt(int x){access(x),splay(x);while(ch[x][0])pushdown(x),x=ch[x][0];return x;} 77 il void mkroot(int x){access(x),splay(x),rev(x);} 78 il void split(int x,int y){mkroot(x),access(y),splay(y);} 79 il void link(int x,int y){mkroot(x),access(y),splay(y),fa[x]=y;} 80 il void cut(int x,int y){split(x,y);fa[x]=ch[y][0]=0,pushup(y);} 81 }lct; 82 int gint() 83 { 84 int rett=0,fh=1;char c=getchar(); 85 while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();} 86 while(c>='0'&&c<='9'){rett=(rett<<3)+(rett<<1)+c-'0';c=getchar();} 87 return rett*fh; 88 } 89 90 int main() 91 { 92 n=gint(),m=gint(); 93 int x,y;ui z;char str[20]; 94 for(int i=1;i<=n;i++) 95 lct.sz[i]=lct.val[i]=lct.sum[i]=lct.mul[i]=1; 96 for(int i=1;i<n;i++) 97 x=gint(),y=gint(),lct.link(x,y); 98 for(int i=1;i<=m;i++) 99 { 100 scanf("%s",str); 101 if(str[0]=='+'){ 102 x=gint(),y=gint(),z=gint(); 103 lct.split(x,y); 104 lct.sum[y]=(lct.sum[y]+(lct.sz[y]*z)%mod)%mod; 105 lct.val[y]=(lct.val[y]+z)%mod; 106 lct.add[y]=(lct.add[y]+z)%mod; 107 }else if(str[0]=='-'){ 108 x=gint(),y=gint(),lct.cut(x,y); 109 x=gint(),y=gint(),lct.link(x,y); 110 }else if(str[0]=='*'){ 111 x=gint(),y=gint(),z=gint(); 112 lct.split(x,y); 113 lct.sum[y]=(lct.sum[y]*z)%mod; 114 lct.val[y]=(lct.val[y]*z)%mod; 115 lct.mul[y]=(lct.mul[y]*z)%mod; 116 lct.add[y]=(lct.add[y]*z)%mod; 117 }else if(str[0]=='/'){ 118 x=gint(),y=gint(),lct.split(x,y); 119 printf("%u ",lct.sum[y]); 120 } 121 } 122 return 0; 123 }