BZOJ_2002_[Hnoi2010]Bounce 彈飛綿羊_LCT

Description

某天，Lostmonkey發明了一種超級彈力裝置，為了在他的綿羊朋友面前顯擺，他邀請小綿羊一起玩個遊戲。遊戲一開始，Lostmonkey在地上沿著一條直線擺上n個裝置，每個裝置設定初始彈力系數ki，當綿羊達到第i個裝置時，它會往後彈ki步，達到第i+ki個裝置，若不存在第i+ki個裝置，則綿羊被彈飛。綿羊想知道當它從第i個裝置起步時，被彈幾次後會被彈飛。為了使得遊戲更有趣，Lostmonkey可以修改某個彈力裝置的彈力系數，任何時候彈力系數均為正整數。

Input

第一行包含一個整數n，表示地上有n個裝置，裝置的編號從0到n-1,接下來一行有n個正整數，依次為那n個裝置的初始彈力系數。第三行有一個正整數m，接下來m行每行至少有兩個數i、j，若i=1，你要輸出從j出發被彈幾次後被彈飛，若i=2則還會再輸入一個正整數k，表示第j個彈力裝置的系數被修改成k。對於20%的數據n,m<=10000，對於100%的數據n<=200000,m<=100000

Output

對於每個i=1的情況，你都要輸出一個需要的步數，占一行。

Sample Input

Sample Output

2
3

每個位置i向i+ki的位置連一條有向邊，如果出去了就向n+1連。

形成了一棵樹，每次詢問相當於求一個點的深度。

這裏直接當成無根樹來做，每次詢問要把n+1變成根。

代碼：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 200050
#define ls ch[p][0]
#define rs ch[p][1]
#define get(x) (ch[f[x]][1]==x)
int a[N],n,f[N],ch[N][2],rev[N],siz[N];
bool isrt(int x) {return ch[f[x]][1]!=x&&ch[f[x]][0]!=x;}
void pushup(int p) {
	if(!p) return ;
	siz[p]=1;
	if(ls) siz[p]+=siz[ls];
	if(rs) siz[p]+=siz[rs];
}
void pushdown(int p) {
	if(rev[p]) {
		rev[ls]^=1; swap(ch[ls][0],ch[ls][1]);
		rev[rs]^=1; swap(ch[rs][0],ch[rs][1]);
		rev[p]=0;
	}
}
void update(int p) {
	if(!isrt(p)) update(f[p]);
	pushdown(p);
}
void rotate(int x) {
	int y=f[x],z=f[y],k=get(x);
	if(!isrt(y)) ch[z][ch[z][1]==y]=x;
	ch[y][k]=ch[x][!k]; f[ch[y][k]]=y;
	ch[x][!k]=y; f[y]=x; f[x]=z;
	pushup(y); pushup(x);
}
void splay(int x) {
	update(x);
	for(int d;d=f[x],!isrt(x);rotate(x))
		if(!isrt(d)) 
			rotate(get(x)==get(d)?d:x);
}
void access(int p) {
	for(int t=0;p;t=p,p=f[p])
		splay(p),rs=t,pushup(p);
}
void makeroot(int p) {
	access(p); splay(p); rev[p]^=1; swap(ls,rs);
}
void link(int x,int p) {
	makeroot(x); splay(p); f[x]=p;
}
void cut(int x,int p) {
	makeroot(x); access(p); splay(p); ls=f[x]=0;
}
int main() {
	scanf("%d",&n);
	int i;
	for(i=1;i<=n+1;i++) siz[i]=1;
	for(i=1;i<=n;i++) {
		scanf("%d",&a[i]);
		link(i,min(n+1,i+a[i]));
	}
	int m,opt,x,y;
	scanf("%d",&m);
	while(m--) {
		scanf("%d%d",&opt,&x); x++;
		if(opt==1) {
			makeroot(n+1); access(x); splay(x); printf("%d\n",siz[ch[x][0]]);
		}else {
			scanf("%d",&y);
			cut(x,min(n+1,x+a[x])); a[x]=y; link(x,min(n+1,x+a[x]));
		}
	}
}

BZOJ_2002_[Hnoi2010]Bounce 彈飛綿羊_LCT