【bzoj2002】[Hnoi2010]Bounce 弹飞绵羊

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec   Memory Limit: 259 MB
Submit: 11597   Solved: 5869
[ Submit][ Status][ Discuss]

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

4
1 2 1 1
3
1 1
2 1 1
1 1

Sample Output

2
3

HINT

Source

[ Submit][ Status][ Discuss]

因为这题特意去学了一下LCT

学长说他签thu靠得就是一道LCT

于是学的时候非常的激动

经过了一个下午的艰苦奋战，LCT终于被攻略

==================================

言归正传

这道题。。。。事实上是LCT裸题

建一个n + 1的点为根，查询的时候查询到根的距离也即splay树上的siz - 1

代码：

#include<cstdio>
#include<vector>
#include<queue>
#include<ctime>
#include<algorithm>
#include<cstdlib>
#include<stack>
#include<cstring>
#include<cmath>
using namespace std;
 
typedef long long LL;

const int INF = 2147483647;
const int maxn = 200100;

int n;
int siz[maxn],ch[maxn][2],fa[maxn],pfa[maxn];

inline LL getint()
{
    LL ret = 0,f = 1;
    char c = getchar();
    while (c < '0' || c > '9')
    {
        if (c == '-') f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9')
        ret = ret * 10 + c - '0',c = getchar();
    return ret * f;
}

inline void maintain(int o)
{
	siz[o] = siz[ch[o][0]] + siz[ch[o][1]] + 1;
}

inline void rotate(int x)
{
	int o = fa[x],y = fa[o];
	pfa[x] = pfa[o]; pfa[o] = 0;
	int d = ch[o][1] == x ? 0 : 1;
	ch[o][d ^ 1] = ch[x][d]; maintain(o);
	if (ch[x][d]) fa[ch[x][d]] = o;
	ch[x][d] = o; maintain(x);
	fa[o] = x; fa[x] = y;
	if (y) ch[y][ch[y][1] == o] = x , maintain(y);
}

inline void splay(int x)
{
	for (int y = fa[x]; y; rotate(x) , y = fa[x])
		if (fa[y]) rotate((ch[y][1] == x) ^ (ch[fa[y]][1] == y) ? x : y);
}

inline void access(int x)
{
	for (int u = x , v = 0; u; v = u,u = pfa[u])
	{
		splay(u);
		if (ch[u][1]) fa[ch[u][1]] = 0 , pfa[ch[u][1]] = u;
		if (v) fa[v] = u , pfa[v] = 0;
		ch[u][1] = v;
		maintain(u);
	}
}

inline void cut(int u)
{
	access(u);
	splay(u);
	if (ch[u][0]) fa[ch[u][0]] = 0 , ch[u][0] = 0;
}

inline void join(int u,int v) //由于只存在一颗树，所以这里并不需要换根 
{
	access(u);
	pfa[v] = u;
	access(v);
}

inline int query(int u)
{
	access(u);
	splay(u);
	return siz[u] - 1;
}

inline void modify(int u,int k)
{
	cut(u);
	join(min(u + k,n + 1),u);
}

int main()
{
	n = getint();
	for (int i = 1; i <= n; i++)
	{
		int k = getint();
		modify(i,k);
	}
	int q = getint();
	for (int i = 1; i <= q; i++)
	{
		int opt = getint();
		if (opt == 1)
		{
			int u = getint() + 1;
			printf("%d\n",query(u));
		}
		else
		{
			int u = getint() + 1,k = getint();
			modify(u,k);
		}
	}
    return 0;
}