2002: [Hnoi2010]Bounce 弹飞绵羊
Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 6908 Solved: 3621
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
LCT裸题。。
直接把x和x+i link,然后最后求子树大小
修改的时候先cut再link
附上本蒟蒻的代码:
#include<cstdio>
#include<iostream>
using namespace std;
#define MAXN 200002
int n,m,father[MAXN],c[MAXN][2],st[MAXN],size[MAXN],next[MAXN];
bool rev[MAXN];
int read()
{
int w=0,f=1; char ch=getchar();
while (ch<'0' || ch>'9')
{
if (ch=='-') f=-1;
ch=getchar();
}
while (ch>='0' && ch<='9')
w=w*10+ch-'0',ch=getchar();
return w*f;
}
bool isroot(int x)
{
return c[father[x]][0]!=x && c[father[x]][1]!=x;
}
void update(int x)
{
int l=c[x][0],r=c[x][1];
size[x]=size[l]+size[r]+1;
}
void pushdown(int x)
{
int l=c[x][0],r=c[x][1];
if (rev[x]) rev[x]^=1,rev[l]^=1,rev[r]^=1,swap(c[x][0],c[x][1]);
}
void rotate(int x)
{
int y=father[x],z=father[y],l,r;
if (c[y][0]==x) l=0;
else l=1;
r=l^1;
if (!isroot(y))
if (c[z][0]==y) c[z][0]=x;
else c[z][1]=x;
father[x]=z,father[y]=x,father[c[x][r]]=y,c[y][l]=c[x][r],c[x][r]=y,update(y),update(x);
}
void splay(int x)
{
int i,top=0,y,z;
st[++top]=x;
for (i=x;!isroot(i);i=father[i]) st[++top]=father[i];
for (i=top;i;i--) pushdown(st[i]);
while (!isroot(x))
{
y=father[x],z=father[y];
if (!isroot(y))
if (c[y][0]==x^c[z][0]==y) rotate(x);
else rotate(y);
rotate(x);
}
}
void access(int x)
{
int t=0;
while (x)
splay(x),c[x][1]=t,t=x,x=father[x];
}
void rever(int x)
{
access(x),splay(x),rev[x]^=1;
}
void cut(int x,int y)
{
rever(x),access(y),splay(y),c[y][0]=father[x]=0;
}
void link(int x,int y)
{
rever(x),father[x]=y,splay(x);
}
int main()
{
int i,x,p,y,t;
n=read();
for (i=1;i<=n;i++)
{
x=read(),father[i]=x+i,size[i]=1;
if (father[i]>n+1) father[i]=n+1;
next[i]=father[i];
}
size[n+1]=1,m=read();
for (i=1;i<=m;i++)
{
p=read();
if (p==1) rever(n+1),x=read(),x++,access(x),splay(x),printf("%d\n",size[c[x][0]]);
if (p==2) x=read(),y=read(),x++,t=min(n+1,x+y),cut(x,next[x]),link(x,t),next[x]=t;
}
return 0;
}