【BZOJ】2002 [Hnoi2010]Bounce 弹飞绵羊 LCT

题目传送门

这题的想法挺好的。至少我想不到。

考虑每个点向它弹到的点连边，如果在当前点被弹飞，就和$n+1$号点连边。

然后发现这样的建图形成了一棵树，修改操作可以看成删边+加边。

一个节点的答案就是该节点到$n+1$号节点的路径长度-1。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=2e5+10;
int n,a[N],m,ch[N][2],f[N],mk[N],sz[N],o,x,y;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

#define isrt(x) ((x)!=ch[f[x]][0]&&(x)!=ch[f[x]][1])
inline void updata(int x){sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;}
inline void rotate(int x){
    int y=f[x],op=(x==ch[y][1]);
    ch[y][op]=ch[x][op^1];
    if (ch[x][op^1]) f[ch[x][op^1]]=y;
    f[x]=f[y];
    if (!isrt(y)) ch[f[y]][ch[f[y]][1]==y]=x;
    f[y]=x,ch[x][op^1]=y;
    return updata(y),updata(x);
}
inline void push(int x){if (mk[x]) mk[x]=0,mk[ch[x][0]]^=1,mk[ch[x][1]]^=1,swap(ch[x][0],ch[x][1]);}
inline void splay(int x){
    static int top,sk[N];sk[top=1]=x;
    for (int i=x; !isrt(i); i=f[i]) sk[++top]=f[i];
    while (top) push(sk[top--]);
    for (int fa=f[x]; !isrt(x); rotate(x),fa=f[x])
        if (!isrt(fa)) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
}
inline void access(int x){for (int lst=0; x; lst=x,x=f[x]) splay(x),ch[x][1]=lst,updata(x);}
inline void root(int x){access(x),splay(x),mk[x]^=1;}
inline void add(int x,int y){root(x),f[x]=y;}
inline void del(int x,int y){root(x),access(y),splay(y),f[x]=ch[y][0]=0,updata(y);}

int main(void){
    read(n);
    for (int i=1; i<=n+1; ++i) sz[i]=1;
    for (int i=1; i<=n; ++i) read(a[i]),add(i,min(n+1,i+a[i]));
    read(m);
    while (m--){
        read(o),read(x),++x;
        if (o==1) root(n+1),access(x),splay(x),printf("%d\n",sz[x]-1);
        else read(y),del(x,min(n+1,x+a[x])),add(x,min(n+1,x+(a[x]=y)));
    }
    return 0;
}