[HNOI2010]弹飞绵羊

一、题目

点此看题

二、解法

从图论角度看，我们可以把$(i,i+a_i)$连边，然后直接从询问的$x$一直往后跳，修改直接改连边。

然后直接写了一个$\text{lct}$，结果样例都过不了，删边和加边都很简单，问题出现在询问，我们的根会因为加边和删边而改变，但我们要从$x$跳出去，不能用$x$和根之间的点数来衡量跳的次数。

那我们建一个超级点$n+1$，把跳出去的点都连到$n+1$，那我们询问$(x,n+1)$之间的点数$-1$即可。

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 200005;
int read()
{
	int x=0,flag=1;char c;
	while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
	while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*flag;
}
int n,m,a[M],par[M],ch[M][2],val[M],sum[M],siz[M],fl[M],st[M];
int nrt(int x)
{
	return ch[par[x]][0]==x || ch[par[x]][1]==x;
}
int chk(int x)
{
	return ch[par[x]][1]==x;
}
void push_up(int x)
{
	if(!x) return ;
	siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
	sum[x]=sum[ch[x][0]]^sum[ch[x][1]]^val[x];
}
void flip(int x)
{
	if(!x) return ;
	swap(ch[x][0],ch[x][1]);
	fl[x]^=1;
}
void push_down(int x)
{
	if(!x) return ;
	if(fl[x])
	{
		flip(ch[x][0]);flip(ch[x][1]);
		fl[x]=0;
	}
}
void rotate(int x)
{
	int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
	ch[y][k]=w;par[w]=y;
	if(nrt(y)) ch[z][chk(y)]=x;par[x]=z;
	ch[x][k^1]=y;par[y]=x;
	push_up(y);push_up(x);
}
void splay(int x)
{
	int y=x,z=0;
	st[++z]=y;
	while(nrt(y)) st[++z]=y=par[y];
	while(z) push_down(st[z--]);
	while(nrt(x))
	{
		int y=par[x],z=par[y];
		if(nrt(y))
		{
			if(chk(y)==chk(x)) rotate(y);
			else rotate(x);
		}
		rotate(x);
	}
}
void access(int x)
{
	for(int y=0;x;x=par[y=x])
		splay(x),ch[x][1]=y,push_up(x); 
}
void makeroot(int x)
{
	access(x);splay(x);
	flip(x);
}
int findroot(int x)
{
	access(x);splay(x);
	while(ch[x][0]) push_down(x),x=ch[x][0];
	splay(x);
	return x;
}
void split(int x,int y)
{
	makeroot(x);
	access(y);splay(y);
}
void link(int x,int y)
{
	makeroot(x);
	if(findroot(y)!=x) par[x]=y;
}
void cut(int x,int y)
{
	makeroot(x);
	if(findroot(y)==x && par[y]==x && !ch[y][0])
	{
		par[y]=ch[x][1]=0;
		push_up(x); 
	}
}
int main()
{
	n=read();
	for(int i=1;i<=n+1;i++)
		siz[i]=1;
	for(int i=1;i<=n;i++)
	{
		a[i]=read();
		if(i+a[i]<=n) link(i,i+a[i]);
		else link(i,n+1);
	}
	m=read();
	while(m--)
	{
		int op=read(),x=read()+1;
		if(op==1)
		{
			makeroot(x);access(n+1);
			splay(n+1);printf("%d\n",siz[n+1]-1);
		}
		if(op==2)
		{
			if(x+a[x]<=n) cut(x,x+a[x]);
			else cut(x,n+1);
			a[x]=read();
			if(x+a[x]<=n) link(x,x+a[x]);
			else link(x,n+1);
		}
	}
}