BZOJ2002：[HNOI2010]弹飞绵羊——题解

http://www.lydsy.com/JudgeOnline/problem.php?id=2002

https://www.luogu.org/problemnew/show/P3203

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

因为它的标签有LCT所以应该是用LCT写的。

因为每个弹射器之间不可能存在环（因为你只能往后跳）所以弹射器和弹飞点（记为n+1）构成了一片森林。

这样我们先link(i,min(i+k[i],n+1)。

对于修改操作显然先cut原先的边在按上面方法link即可。

对于查询我们就是相当于求n+1和i的路径长，先makeroot(n+1)，再access(i)，这样n+1到i的实边长度即为所求，然后求实链所代表的平衡树大小-1即可。

（下面为吐槽）

说实话最开始没想到LCT怎么求两点最短路。

但是后来一想这不就是access一下之后求实边长度吗。

再看了洛谷超易懂题解发现实链不就是一棵平衡树吗，求一遍平衡树大小-1不就得了？

（假装这题很简单）

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cctype>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
const int N=2e5+;
int n,m,r,k[N],fa[N],tr[N][],rev[N],q[N],size[N];
inline int read(){
    int X=,w=;char ch=;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
    return w?-X:X;
}
inline bool get(int x){
    return tr[fa[x]][]==x;
}
inline bool isroot(int x){
    if(!fa[x])return ;
    return tr[fa[x]][]!=x&&tr[fa[x]][]!=x;
}
inline void upt(int x){
    size[x]=;
    if(tr[x][])size[x]+=size[tr[x][]];
    if(tr[x][])size[x]+=size[tr[x][]];
}
inline void pushrev(int x){
    if(!rev[x])return;
    swap(tr[x][],tr[x][]);
    if(tr[x][])rev[tr[x][]]^=;
    if(tr[x][])rev[tr[x][]]^=;
    rev[x]=;
}
inline void rotate(int x){
    int y=fa[x],z=fa[y],b=tr[y][]==x?tr[x][]:tr[x][];
    if(z&&!isroot(y))(tr[z][]==y?tr[z][]:tr[z][])=x;
    fa[x]=z;fa[y]=x;b?fa[b]=y:;
    if(tr[y][]==x)tr[x][]=y,tr[y][]=b;
    else tr[x][]=y,tr[y][]=b;
    upt(y);upt(x);
}
inline void splay(int x){
    q[r=]=x;
    for(int y=x;!isroot(y);y=fa[y])q[++r]=fa[y];
    for(int i=r;i>=;i--)pushrev(q[i]);
    while(!isroot(x)){
    if(!isroot(fa[x]))
        rotate((get(x)==get(fa[x])?fa[x]:x));
    rotate(x);
    }
    upt(x);
}
inline void access(int x){
    for(int y=;x;y=x,x=fa[x]){
    splay(x);tr[x][]=y;
    if(y)fa[y]=x;
    }
}
inline int findroot(int x){
    access(x);splay(x);
    while(pushrev(x),tr[x][])x=tr[x][];
    splay(x);
    return x;
}
inline void makeroot(int x){
    access(x);splay(x);
    rev[x]^=;
}
inline void link(int x,int y){
    makeroot(x);fa[x]=y;
}
inline void cut(int x,int y){
    makeroot(x);access(y);splay(y);
    tr[y][]=;fa[x]=;
}
int main(){
    n=read();
    for(int i=;i<=n;i++)
    link(i,min(i+(k[i]=read()),n+));
    m=read();
    for(int i=;i<=m;i++){
    int op=read(),j=read()+;
    if(op==){
        makeroot(j);access(n+);splay(n+);
        printf("%d\n",size[n+]-);
    }else{
        cut(j,min(j+k[j],n+));
        link(j,min(j+(k[j]=read()),n+));
    }
    }
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++