BZOJ 3091: 城市旅行 lct 期望 splay

https://www.lydsy.com/JudgeOnline/problem.php?id=3091

https://blog.csdn.net/popoqqq/article/details/40823659

看题解吧,没什么好解释的。。。。板子题,

我觉得以后我写lct都可以像这样专门写个rev和add的函数出来,很好用。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<algorithm>
  4 #include<cstring>
  5 #include<cmath>
  6 #include<queue>
  7 using namespace std;
  8 const int maxn=50010;
  9 int n,m;
 10 int fa[maxn]={},ch[maxn][2]={};
 11 long long siz[maxn]={},val[maxn]={},sum[maxn]={};
 12 long long lsum[maxn]={},rsum[maxn]={};
 13 long long num[maxn]={};long long ad[maxn]={};
 14 int rev[maxn]={};int sta[maxn]={},tail=0;
 15 inline void updata(int x){
 16     sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+val[x];
 17     siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
 18     
 19     lsum[x]=lsum[ch[x][0]]+lsum[ch[x][1]]+(sum[ch[x][1]]+val[x])*(siz[ch[x][0]]+1);
 20     rsum[x]=rsum[ch[x][1]]+rsum[ch[x][0]]+(sum[ch[x][0]]+val[x])*(siz[ch[x][1]]+1);
 21     
 22     num[x]=num[ch[x][0]]+num[ch[x][1]]
 23     +lsum[ch[x][0]]*(siz[ch[x][1]]+1)
 24     +rsum[ch[x][1]]*(siz[ch[x][0]]+1)
 25     +val[x]*(siz[ch[x][0]]+1)*(siz[ch[x][1]]+1);
 26 }
 27 inline long long hex(long long x){return x*(x+1)/2;}
 28 inline long long hex2(long long x){return x*(x+1)*(x+2)/6;}
 29 void Add(int x,long long v){
 30     ad[x]+=v; sum[x]+=siz[x]*v; val[x]+=v;
 31     lsum[x]+=hex(siz[x])*v;
 32     rsum[x]+=hex(siz[x])*v;
 33     num[x]+=hex2(siz[x])*v;
 34 }
 35 void Rev(int x){
 36     swap(ch[x][0],ch[x][1]);
 37     swap(lsum[x],rsum[x]);
 38     rev[x]^=1;
 39 }
 40 inline void downdata(int x){
 41     if(rev[x]){
 42         if(ch[x][0])Rev(ch[x][0]);
 43         if(ch[x][1])Rev(ch[x][1]);
 44         updata(x);
 45         rev[x]=0;
 46     }
 47     if(ad[x]){
 48         if(ch[x][0])Add(ch[x][0],ad[x]);
 49         if(ch[x][1])Add(ch[x][1],ad[x]);
 50         updata(x);
 51         ad[x]=0;
 52     }
 53 }
 54 inline bool isroot(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;}
 55 void rotate(int x){
 56     int y=fa[x];int fy=fa[y];
 57     int l=ch[y][0]==x?0:1;int r=l^1;
 58     if(!isroot(y)){
 59         if(ch[fy][0]==y)ch[fy][0]=x;
 60         else ch[fy][1]=x;
 61     }
 62     fa[ch[x][r]]=y;fa[y]=x;fa[x]=fy;
 63     ch[y][l]=ch[x][r];ch[x][r]=y;
 64     updata(y);
 65 }
 66 void Splay(int x){
 67     int y,fy,w=x;sta[++tail]=x;
 68     while(!isroot(w)){ sta[++tail]=fa[w]; w=fa[w]; }
 69     while(tail)downdata(sta[tail--]);
 70     while(!isroot(x)){
 71         y=fa[x];fy=fa[y];
 72         if(!isroot(y)){
 73             if((ch[fy][0]==y)^(ch[y][0]==x)) rotate(x);
 74             else rotate(y);
 75         }
 76         rotate(x);
 77     }updata(x);
 78 }
 79 void Access(int x){
 80     int y=0;
 81     while(x){
 82         Splay(x);ch[x][1]=y;
 83         updata(x);
 84         y=x;x=fa[y];
 85     }
 86 }
 87 int findfa(int x){
 88     Access(x);Splay(x);
 89     while(ch[x][0])x=ch[x][0];
 90     return x;
 91 }
 92 void Reverse(int x){
 93     Access(x);Splay(x);
 94     Rev(x);
 95 }
 96 void Cut(int x,int y){
 97     Reverse(x);Access(y);Splay(y);
 98     if(ch[y][0]==x&&!ch[x][1]){fa[x]=0;ch[y][0]=0;updata(y);}
 99 }
100 void Link(int x,int y){
101     Reverse(x);
102     fa[x]=y;
103 }
104 long long mgcd(long long x,long long y){
105     long long z;
106     while(y){
107         z=x%y;x=y;y=z;
108     }return x;
109 }
110 int main(){
111     //freopen("a.in","r",stdin);
112     //freopen("a.out","w",stdout);
113     scanf("%d%d",&n,&m);
114     for(int i=1;i<=n;i++){scanf("%lld",&val[i]);siz[i]=1;sum[i]=lsum[i]=rsum[i]=num[i]=val[i];}
115     int x,y,op;long long v,aa,bb;
116     for(int i=1;i<n;i++){
117         scanf("%d%d",&x,&y); if(findfa(x)!=findfa(y))Link(x,y);
118     }
119     for(int i=1;i<=m;i++){
120         scanf("%d%d%d",&op,&x,&y);
121         if(op==1){
122             if(findfa(x)==findfa(y))Cut(x,y);
123         }
124         else if(op==2){
125             if(findfa(x)!=findfa(y))Link(x,y);
126         }
127         else if(op==3){
128             scanf("%lld",&v);
129             if(findfa(x)==findfa(y)){
130                 Reverse(x);Access(y);Splay(y);
131                 Add(y,v);
132             }
133         }
134         else{
135             if(findfa(x)!=findfa(y)){printf("-1
");}
136             else{
137                 Reverse(x);Access(y);Splay(y);
138                 aa=num[y];bb=hex(siz[y]);
139                 v=mgcd(aa,bb);
140                 printf("%lld/%lld
",aa/v,bb/v);
141             }
142         }
143     }
144 }
View Code

原文地址:https://www.cnblogs.com/137shoebills/p/8661650.html