BZOJ 2631 [国家集训队]Tree II (LCT)

题目大意:给你一棵树,让你维护一个数据结构,支持

边的断,连

树链上所有点点权加上某个值

树链上所有点点权乘上某个值

求树链所有点点权和

 

(辣鸡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\n",lct.sum[y]);
120         }
121     }
122     return 0;
123 }

 

posted @ 2018-09-24 22:36  guapisolo  阅读(229)  评论(0编辑  收藏  举报