洛谷 3203 HNOI2010 BOUNCE 弹飞绵羊

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define N (500010)
 4 #define ls (c[u][0])
 5 #define rs (c[u][1])
 6 using namespace std;
 7 int n,m,opt,k[N];
 8 inline int read(){
 9     int k=0,f=1; char c=getchar();
10     while(c<‘0‘||c>‘9‘)c==‘-‘&&(f=-1),c=getchar();
11     while(‘0‘<=c&&c<=‘9‘)k=k*10+c-‘0‘,c=getchar();
12     return k*f;
13 }
14 struct Link_Cut_Tree{
15     int top,c[N][2],fa[N],rev[N],size[N],q[N];
16     inline void pushdown(int u){
17         if(rev[u]) rev[ls]^=1,rev[rs]^=1,rev[u]^=1,swap(ls,rs);
18     }
19     inline void pushup(int u){
20         size[u]=1;
21         if(ls) size[u]+=size[ls];
22         if(rs) size[u]+=size[rs];
23     }
24     inline bool isroot(int u){
25         return c[fa[u]][0]!=u&&c[fa[u]][1]!=u;
26     }
27     inline bool which(int u){
28         return c[fa[u]][1]==u;
29     }
30     void rotate(int u){
31         int f=fa[u],gf=fa[f],wh=which(u);
32         if(!isroot(f)) c[gf][which(f)]=u;
33         fa[u]=gf; fa[f]=u; fa[c[u][wh^1]]=f;
34         c[f][wh]=c[u][wh^1]; c[u][wh^1]=f;
35         pushup(f); pushup(u);
36     }
37     void splay(int u){
38         q[top=1]=u;
39         for(int i=u;!isroot(i);i=fa[i]) q[++top]=fa[i];
40         for(int i=top;i;i--) pushdown(q[i]);
41         while(!isroot(u)){
42             if(!isroot(fa[u])) rotate(which(u)==which(fa[u])?fa[u]:u);
43             rotate(u);
44         }
45         pushup(u);
46     }
47     void access(int u){
48         for(int son=0;u;son=u,u=fa[u]) 
49             splay(u),c[u][1]=son,pushup(u);
50     }
51     void makeroot(int u){
52         access(u); splay(u); rev[u]^=1;
53     }
54     int find(int u){
55         access(u); splay(u);
56         while(ls) u=ls; splay(u); 
57         return u;
58     }
59     void split(int x,int y){
60         makeroot(x); access(y); splay(y);
61     }
62     void cut(int x,int y){
63         int xrt=find(x),yrt=find(y);
64         if(xrt!=yrt) return;
65         split(x,y);
66         if(c[y][0]==x) c[y][0]=0,fa[x]=0; 
67         pushup(y);
68     }
69     void link(int x,int y){
70         int xrt=find(x),yrt=find(y);
71         if(xrt==yrt) return;
72         makeroot(x); fa[x]=y;
73     }
74 }t;
75 int main(){
76     n=read();
77     for(int i=1;i<=n;i++) t.link(i,min(n+1,i+(k[i]=read())));    
78     m=read();
79     while(m--){
80         if((opt=read())==1){
81             int x=read()+1;
82             t.makeroot(x); t.access(n+1); t.splay(n+1);
83             printf("%d\n",t.size[n+1]-1);
84         }
85         else{
86             int x=read()+1,y=read();
87             if(x+y>n+1&&x+k[x]>n+1) continue;
88             t.cut(x,min(n+1,x+k[x]));
89             t.link(x,min(n+1,x+y));
90             k[x]=y;
91         }
92     }
93     return 0;
94 }