题目描述
某天,Lostmonkey发明了一种超级弹力装置,为了在他的绵羊朋友面前显摆,他邀请小绵羊一起玩个游戏。游戏一开始,Lostmonkey在地上沿着一条直线摆上n个装置,每个装置设定初始弹力系数ki,当绵羊达到第i个装置时,它会往后弹ki步,达到第i+ki个装置,若不存在第i+ki个装置,则绵羊被弹飞。绵羊想知道当它从第i个装置起步时,被弹几次后会被弹飞。为了使得游戏更有趣,Lostmonkey可以修改某个弹力装置的弹力系数,任何时候弹力系数均为正整数。
题解:
LCT。我们把每个点向它跳到的点连边,如果超过了n,就不连了,修改的话可以写link和cut,但是我写了一种许多原本LCT需要的函数都没写的写法,这个的link和cut可以直接改。
讲解一下核心代码吧。
核心代码是access(x);splay(x);
这个东西就是你把x到根的路径连通,之后再把x旋转到根,这样原来x与儿子连的边变成了轻边,不会被统计,所有原来比x深度小的点都到了x的左子树中,而此时x所在的子树大小就是步数。
代码如下:
#include <bits/stdc++.h>
using namespace std;
int n,m,a,s[200001],f[200001],c[200001][2],rev[200001];
int nroot(int x)
{
return c[f[x]][0]==x||c[f[x]][1]==x;
}
void pushup(int x)
{
s[x]=s[c[x][1]]+s[c[x][0]]+1;
}
void rotate(int x)
{
int y=f[x],z=f[y],k=c[y][1]==x,w=c[x][!k];
if(nroot(y))
c[z][c[z][1]==y]=x;
c[x][!k]=y;
c[y][k]=w;
if(w)
f[w]=y;
f[x]=z;
f[y]=x;
pushup(y);
}
void splay(int x)
{
int y,z;
while(nroot(x))
{
y=f[x];
z=f[y];
if(nroot(y))
{
if(c[y][0]==x ^ c[z][0]==y)
rotate(y);
else
rotate(x);
}
rotate(x);
}
pushup(x);
}
void access(int x)
{
int y=0;
while(x!=0)
{
splay(x);
c[x][1]=y;
pushup(x);
if(y!=0)
f[y]=x;
y=x;
x=f[x];
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
s[i]=1;
if(a+i<=n)
f[i]=a+i;
}
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d",&x);
if(x==1)
{
scanf("%d",&y);
++y;
access(y);
splay(y);
printf("%d\n",s[y]);
}
if(x==2)
{
scanf("%d%d",&y,&z);
++y;
access(y);
splay(y);
c[y][0]=f[c[y][0]]=0;
if(y+z<=n)
f[y]=y+z;
}
}
return 0;
}