BZOJ2002 洛谷3203 HNOI2010 弹飞绵羊LCT

题目描述

某天，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;
}