bzoj2631 tree

题目链接

Link cut tree

维护size和两个标记即可

黄学长告诉我long long 会T……

上面long long 下面unsigned……

  1 #include<algorithm>
  2 #include<iostream>
  3 #include<cstdlib>
  4 #include<cstring>
  5 #include<cstdio>
  6 #include<string>
  7 #include<cmath>
  8 #include<ctime>
  9 #include<queue>
 10 #include<stack>
 11 #include<map>
 12 #include<set>
 13 #define rre(i,r,l) for(int i=(r);i>=(l);i--)
 14 #define re(i,l,r) for(int i=(l);i<=(r);i++)
 15 #define Clear(a,b) memset(a,b,sizeof(a))
 16 #define inout(x) printf("%d",(x))
 17 #define douin(x) scanf("%lf",&x)
 18 #define strin(x) scanf("%sum",(x))
 19 #define LLin(x) scanf("%lld",&x)
 20 #define op operator
 21 #define CSC main
 22 typedef unsigned long long ULL;
 23 typedef const int cint;
 24 typedef long long LL;
 25 using namespace std;
 26 cint mod=51061;
 27 void inin(int &ret)
 28 {
 29     ret=0;int f=0;char ch=getchar();
 30     while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
 31     while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
 32     ret=f?-ret:ret;
 33 }
 34 int ch[100010][2],rev[100010],fa[100010];
 35 unsigned w[100010],sum[100010],add[100010],cheng[100010],s[100010];
 36 bool isroot(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;}
 37 void maintain(int x)
 38 {
 39     if(x)
 40     {
 41         sum[x]=(w[x]+sum[ch[x][0]]+sum[ch[x][1]])%mod;
 42         s[x]=1+s[ch[x][0]]+s[ch[x][1]];
 43     }
 44 }
 45 void addtag(int x,int a,int b)
 46 {
 47     w[x]=(w[x]*a+b)%mod;
 48     sum[x]=(sum[x]*a+b*s[x])%mod;
 49     add[x]=(add[x]*a+b)%mod;
 50     cheng[x]=cheng[x]*a%mod;
 51 }
 52 void rotate(int x)
 53 {
 54     int y=fa[x],z=fa[y];
 55     if(!isroot(y))ch[z][ch[z][1]==y]=x;
 56     fa[x]=z,fa[y]=x;
 57     int d=ch[y][1]==x;
 58     fa[ch[x][d^1]]=y;
 59     ch[y][d]=ch[x][d^1];
 60     ch[x][d^1]=y;
 61     maintain(y),maintain(x);
 62 }
 63 void down(int x)
 64 {
 65     if(rev[x])
 66     {
 67         swap(ch[x][0],ch[x][1]);
 68         rev[ch[x][0]]^=1;
 69         rev[ch[x][1]]^=1;
 70         rev[x]=0;
 71     }
 72     if(cheng[x]!=1||add[x]!=0)
 73     {
 74         addtag(ch[x][0],cheng[x],add[x]);
 75         addtag(ch[x][1],cheng[x],add[x]);
 76     }
 77     add[x]=0;cheng[x]=1;
 78 }
 79 int sta[100010],top;
 80 void splay(int x)
 81 {
 82     top=0;int xx=x;sta[++top]=xx;
 83     while(!isroot(xx))sta[++top]=fa[xx],xx=fa[xx];
 84     while(top)down(sta[top--]);
 85     while(!isroot(x))
 86     {
 87         int y=fa[x],z=fa[y];
 88         if(!isroot(y))
 89             if((ch[y][1]==x)^(ch[z][1]==y))rotate(x);
 90             else rotate(y);else ;
 91         rotate(x);
 92     }
 93 }
 94 void access(int x)
 95 {
 96     int temp=0;
 97     while(x)
 98     {
 99         splay(x);
100         ch[x][1]=temp;
101         maintain(x);
102         temp=x,x=fa[x];
103     }
104 }
105 void reverse(int x)
106 {
107     access(x),splay(x),rev[x]^=1;
108 }
109 void link(int x,int y)
110 {
111     reverse(x),fa[x]=y,splay(x);
112 }
113 void cut(int x,int y)
114 {
115     reverse(x),access(y),splay(y),ch[y][0]=fa[x]=0;maintain(y);
116 }
117 int n,m;
118 int main()
119 {
120     inin(n),inin(m);
121     re(i,1,n)s[i]=w[i]=sum[i]=cheng[i]=1;
122     re(i,2,n)
123     {
124         int x,y;
125         inin(x),inin(y);
126         link(x,y);
127     }
128     char ss[5];
129     re(i,1,m)
130     {
131         strin(ss);int a,b,c,d;
132         if(ss[0]=='+')
133         {
134             inin(a),inin(b),inin(c);
135             reverse(a),access(b),splay(b);
136             addtag(b,1,c);
137         }
138         else if(ss[0]=='-')
139         {
140             inin(a),inin(b),inin(c),inin(d);
141             cut(a,b),link(c,d);
142         }
143         else if(ss[0]=='*')
144         {
145             inin(a),inin(b),inin(c);
146             reverse(a),access(b),splay(b);
147             addtag(b,c,0);
148         }
149         else 
150         {
151             inin(a),inin(b);
152             reverse(a),access(b),splay(b);
153             printf("%d
",sum[b]);
154         }
155     }
156      return 0;
157 }
原文地址:https://www.cnblogs.com/HugeGun/p/5237220.html