【bzoj2002】[Hnoi2010]Bounce 弹飞绵羊

题目描述：

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

输入：
第一行包含一个整数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

输出：
对于每个i=1的情况，你都要输出一个需要的步数，占一行。

样例输入：
4
1 2 1 1
3
1 1
2 1 1
1 1

样例输出：

2
3

题解：

把每个位置看成一个点，每个点往后跳的位置看做是树的边，新建一个点表示被弹飞的点（根节点）。你需要支持修改边，查询某个点的深度。lct果题。（第一次写lct）

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

#ifdef WIN32
	#define LL "%I64d"
#else
	#define LL "%lld"
#endif

#ifdef CT
	#define debug(...) printf(__VA_ARGS__)
	#define setfile() 
#else
	#define debug(...)
	#define filename ""
	#define setfile() freopen(filename".in", "r", stdin); freopen(filename".out", "w", stdout);
#endif

#define R register
#define getc() (S == T && (T = (S = B) + fread(B, 1, 1 << 15, stdin), S == T) ? EOF : *S++)
#define dmax(_a, _b) ((_a) > (_b) ? (_a) : (_b))
#define dmin(_a, _b) ((_a) < (_b) ? (_a) : (_b))
#define cmax(_a, _b) (_a < (_b) ? _a = (_b) : 0)
#define cmin(_a, _b) (_a > (_b) ? _a = (_b) : 0)
char B[1 << 15], *S = B, *T = B;
inline int FastIn()
{
	R char ch; R int cnt = 0; R bool minus = 0;
	while (ch = getc(), (ch < '0' || ch > '9') && ch != '-') ;
	ch == '-' ? minus = 1 : cnt = ch - '0';
	while (ch = getc(), ch >= '0' && ch <= '9') cnt = cnt * 10 + ch - '0';
	return minus ? -cnt : cnt;
}
#define maxn 200010
struct Node *null;
struct Node
{
	Node *ch[2], *fa;
	int size;
	inline void update()
	{
		size = ch[0] -> size + ch[1] -> size + 1;
	}
	inline bool type()
	{
		return fa -> ch[1] == this;
	}
	inline bool check()
	{
		return fa -> ch[type()] == this;
	}
	inline void rotate()
	{
		R Node *f = fa;
		R bool d = type();
		(fa = f -> fa), f -> check() ? fa -> ch[f -> type()] = this : 0;
		(f -> ch[d] = ch[!d]) != null ? ch[!d] -> fa = f : 0;
		f -> fa = this;
		ch[!d] = f;
		f -> update();
	}
	inline void splay()
	{
		for (; check(); rotate())
			if (fa -> check())
				(type() != fa -> type() ? this : fa) -> rotate();
		update();
	}
	inline Node *access()
	{
		R Node *i = this, *j = null;
		for (; i != null; i = (j = i) -> fa)
		{
			i -> splay();
			i -> ch[1] = j;
			i -> update();
		}
		return j;
	}
	inline void make_root()
	{
		access();
		splay();
	}
	inline void link(R Node *that)
	{
		make_root();
		ch[0] -> fa = null;
		ch[0] = null;
		fa = that;
		update();
	}
}mem[maxn], *tot = mem, *fir = mem;
int main()
{
//	setfile();
	null = mem;
	null -> size = 0;
	null -> fa = null -> ch[0] = null -> ch[1] = null;
	R int n = FastIn();
	for (R int i = 1; i <= n; ++i) *++tot = (Node) {null, null, null, 1};
	for (R int i = 1; i <= n; ++i)
	{
		R int x = FastIn();
		if (i + x <= n) (mem + i) -> fa = mem + i + x;
	}
	R int m = FastIn();
	for (R int i = 1; i <= m; ++i)
	{
		R int opt = FastIn(), x = FastIn() + 1;
		if (opt == 1)
		{
			R Node *now = fir + x;
			now -> make_root();
			printf("%d\n",now -> size );
		}
		else
		{
			R int k = FastIn();
			R Node *now = fir + x;
			now -> link(x + k > n ? null : now + k);
		}
	}
	return 0;
}
/*
4
1 2 1 1
3
1 1
2 1 1
1 1
*/