BZOJ 2002: [Hnoi2010]Bounce 彈飛綿羊(LCT裸題)

題目連接：BZOJ 2002: [Hnoi2010]Bounce 彈飛綿羊

題意：

有n個節點，每一個節點有個彈力系數x_i，表示第i個節點能夠彈到i+x_i，若是超過了n就彈飛。

如今有兩個操做：

1 x 詢問從x開始要彈幾回才能彈飛。

2 x y 將x的彈力系數更改成y。

題解：

這題能夠分塊能夠LCT。

不過nlongn的均攤LCT比nsqrt(n)的分塊慢，由於分塊的常數小。

這裏我用LCT。

考慮新加一個節點n+1,表示爲彈飛的節點。

每一個節點只會彈向一個節點，因此能夠構建一棵樹，

因爲LCT的splay是按深度爲關鍵字的，因此輸出左子樹的大小就好了。(注意輸出前要先將n+1設爲根)

 1 #include<bits/stdc++.h>
 2 #define F(i,a,b) for(int i=a;i<=b;i++)
 3 #define mst(a,b) memset(a,b,sizeof(a))
 4 using namespace std;
 5 
 6 namespace LCT
 7 {
 8     const int N=2e5+7;
 9     int f[N],son[N][2],val[N],sum[N],tmp[N],lazy[N];
10     int g[N],v[N*2],nxt[N*2],ed;bool rev[N];
11     void clear(int n)
12     {
13         F(i,1,n)f[i]=son[i][0]=son[i][1]=0;
14         F(i,1,n)rev[i]=lazy[i]=g[i]=0;ed=0;
15     }
16     void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
17     void build(int x=1){
18         sum[x]=val[x];
19         for(int i=g[x];i;i=nxt[i])
20             if(!f[v[i]]&&v[i]!=1)f[v[i]]=x,build(v[i]);    
21     }
22     bool isroot(int x){return !f[x]||son[f[x]][0]!=x&&son[f[x]][1]!=x;}
23     void rev1(int x){if(!x)return;swap(son[x][0],son[x][1]);rev[x]^=1;}
24     void add(int x,int c){if(!x)return;sum[x]+=c,val[x]+=c,lazy[x]+=c;}
25     void pb(int x){
26         if(rev[x])rev1(son[x][0]),rev1(son[x][1]),rev[x]=0;
27         if(lazy[x])add(son[x][0],lazy[x]),add(son[x][1],lazy[x]),lazy[x]=0;
28     }
29     void up(int x){
30         sum[x]=val[x];
31         if(son[x][0])sum[x]+=sum[son[x][0]];
32         if(son[x][1])sum[x]+=sum[son[x][1]];
33     }
34     void rotate(int x){
35         int y=f[x],w=son[y][1]==x;
36         son[y][w]=son[x][w^1];
37         if(son[x][w^1])f[son[x][w^1]]=y;
38         if(f[y]){
39             int z=f[y];
40             if(son[z][0]==y)son[z][0]=x;else if(son[z][1]==y)son[z][1]=x;
41         }
42         f[x]=f[y];f[y]=x;son[x][w^1]=y;up(y);
43     }
44     void splay(int x){
45         int s=1,i=x,y;tmp[1]=i;
46         while(!isroot(i))tmp[++s]=i=f[i];
47         while(s)pb(tmp[s--]);
48         while(!isroot(x)){
49             y=f[x];
50             if(!isroot(y)){if((son[f[y]][0]==y)^(son[y][0]==x))rotate(x);else rotate(y);}
51             rotate(x);
52         }
53         up(x);
54     }
55     void access(int x){for(int y=0;x;y=x,x=f[x])splay(x),son[x][1]=y,up(x);}
56     int root(int x){access(x);splay(x);while(son[x][0])x=son[x][0];return x;}
57     void makeroot(int x){access(x);splay(x);rev1(x);}
58     void link(int x,int y){makeroot(x);f[x]=y;access(x);}
59     void cutf(int x){access(x);splay(x);f[son[x][0]]=0;son[x][0]=0;up(x);}
60     void cut(int x,int y){makeroot(x);cutf(y);}
61     void update(int x,int y,int c){makeroot(x),access(y),splay(y),add(y,c);}
62     int ask(int x,int y){makeroot(x);access(y);splay(y);return sum[y];}
63 }
64 using namespace LCT;
65 int n,m,x,y,op;
66 
67 int main()
68 {
69     scanf("%d",&n);
70     F(i,1,n+1)val[i]=1;
71     F(i,1,n)
72     {
73         scanf("%d",&x);
74         if(x+i<=n)f[i]=x+i;
75         else f[i]=n+1;
76     }
77     scanf("%d",&m);
78     F(i,1,m)
79     {
80         scanf("%d",&op);
81         if(op==1)
82         {
83             scanf("%d",&x),x++;
84             makeroot(n+1);
85             access(x);splay(x);
86             printf("%d\n",sum[son[x][0]]);
87         }else
88         {
89             scanf("%d%d",&x,&y),x++;
90             makeroot(n+1),cutf(x);
91             if(x+y<=n)link(x,x+y);
92             else link(x,n+1);
93         }
94     }
95     return 0;
96 }

View Code