[HNOI2010]BOUNCE 弹飞绵羊

[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的情况，你都要输出一个需要的步数，占一行。

输入样例#1：

4

1 2 1 1

3

1 1

2 1 1

1 1

输出样例#1：

2

3

题解：

本来是向用这道题好好练一练lct的，没想到卡了一天，最后还是对着板子跳出来的，找出了一堆错。。。。。

思路应该不难想到吧，新建一个root节点表示弹飞之后到的点。每次更改弹力系数，先将原边删掉，如果w[i]+i<=n，和w[i]+i,若大于n则，和root连一条边。对于每个查询，查询当前节点到root的size值就可以了。注意cut的时候也要加上对是否大于n的判断！！

还是那句话，好好背模板才是王道。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;

const int max_n = 2000010;

struct splay_node{
	int size,value,rev,pos;//根据题意，维护size就可以了 
	splay_node *ch[2];
	splay_node *pre;
	splay_node *the;//没什么用处（划掉）好调，自动忽略 
	
	void update();
	void pushdown();
	int get_wh();
	void set_ch(splay_node *child,int wh)
	{
		pushdown();
	    ch[wh]=child;
	    child->pre=this;
	    update();
	};
	bool is_root(); 
}pool[max_n],*null,*root;

int w[max_n];
int n,x,y,tot,c,m;

inline void splay_node::update()
{
	size=1;
	if(ch[0]!=null) size+=ch[0]->size;
	if(ch[1]!=null) size+=ch[1]->size;
}

inline void splay_node::pushdown()
{
	if(rev)
	{
		rev=0;
		swap(ch[0],ch[1]);
		ch[0]->rev^=1;
		ch[1]->rev^=1;
	}
}

inline int splay_node::get_wh()
{
	return pre->ch[0]==this? 0 : 1;
}

inline bool splay_node::is_root()
{
	if(pre==null) return true;
	return (pre->ch[0]!=this && pre->ch[1]!=this);
}

inline void init()
{
	tot=0;
	null=pool;
	null->size=0;
	null->rev=0;
	null->value=0;
	null->ch[0]=null->ch[1]=null->pre=null;
	null->the=null;
	
	root=pool+n+1;
	root->size=1;
	root->rev=0;
	root->value=0;
	root->ch[0]=root->ch[1]=root->pre=null;
	root->the=root;
}

inline splay_node *get_new(int val)
{
	splay_node *newone;
	newone = pool+ ++tot;
	newone->size=1;
	newone->rev=0;
	newone->value=val;
	newone->ch[0]=newone->ch[1]=newone->pre=null;
	newone->the=newone;
	return newone;
}

inline void rotate(splay_node *now)
{	
    splay_node *fa=now->pre,*grand=now->pre->pre;
	if(grand!=null) grand->pushdown();
	fa->pushdown(); now->pushdown();
	int wh=now->get_wh();
	fa->set_ch(now->ch[wh^1],wh);
	if(fa->is_root()) now->pre=grand;
	else grand->set_ch(now,fa->get_wh());
	now->set_ch(fa,wh^1);
	fa->update(); now->update();
	if(grand!=null) grand->update();
}

inline void splay(splay_node *now)
{
	if(now->is_root()) return;
	for(; !now->is_root(); rotate(now))
	  if(!now->pre->is_root())
	    now->pre->get_wh()==now->get_wh() ? rotate(now->pre) : rotate(now);
}

inline splay_node *access(splay_node *now)
{
 	splay_node *y=null;
	
	for(; now!=null; y=now,now=now->pre)
	{
		splay(now);
		now->set_ch(y,1);
	}
	
	return y;
}

inline void changeroot(splay_node *now)
{
    access(now)->rev^=1;
    splay(now);
}

inline void link(splay_node *u,splay_node *v)
{
	changeroot(u);
	u->pre=v;
	access(u);
}

inline void cut(splay_node *u,splay_node *v)
{
	changeroot(u);
	access(v);
	splay(u);
	u->pushdown();
	u->ch[1]=v->pre=null;
	u->update();
}

inline int query(splay_node *u,splay_node *v)
{
	changeroot(u);
	access(v);
	splay(v);
	return v->size;
}

int main()
{
	scanf("%d",&n);
	init();
	
	for(int i=1; i<=n; ++i)
	{
		scanf("%d",&w[i]);
		get_new(w[i])->pos=i;
	}
	
	for(int i=1; i<=n; ++i)
	{
		if(w[i]+i<=n) link(pool+i,pool+i+w[i]);
	    else link(pool+i,root);
	}
	
	scanf("%d",&m);
	
	for(int i=1; i<=m; ++i)
	{
		scanf("%d",&c);
		
		if(c==1)
		{
			scanf("%d",&x);
			x+=1;
			printf("%d\n",query(root,pool+x)-1);
		}
		else
		{
			scanf("%d%d",&x,&y);
			x+=1;
			if(w[x]+x<=n)//注意这里的判断，之前一直RE 
			  cut(pool+x,pool+w[x]+x);
			else cut(pool+x,root);
			w[x]=y;
			if(w[x]+x<=n && w[x]!=0) link(pool+x,pool+w[x]+x);
			else link(pool+x,root);
		}
	}
	
	return 0;
}