[bzoj2002][动态树]弹飞绵羊

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 13160 Solved: 6694
[Submit][Status][Discuss]
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

HINT

Source

动态树：

因为lct这个东西我一直都掌握的不熟练，今天挑出来重点理一理。
以下只讲实现和实现的原理及思路，不考虑证明过程。
注意一下提到的根都是splay的根，树的根会特意提为树根,因为lct维护的是动态树,其实就是森林,每个树往往没有树根,需要由我们自己指定,接下来的内容将先不涉及指定树根,大家可以先假定树根为1且图连通为一棵树。
如果树是静态的，我们有很好的树链剖分技巧来实现大多静态树上的操作。但是有些问题中涉及到了加边删边的操作，这时我们就要用到lct了。lct的主要思想就是动态的维护树剖分。
说到树剖分，我们显然要先给出一种树的剖分方式。在lct中，一个节点x，他子树中最后一个被访问到的点是y，那么x到y就是一条重链，换句话说，一个重链的顶端，他会一直延伸到这条链最后一个被访问到的点。这就是lct的剖分方式。
那我们如何动态的维护这种剖分方式呢？朴素做法是我们对于一个被访问到的点，我们暴力的跳他的祖先，然后暴力的把他祖先的链给修改成正确的信息。但是这种做法实在太暴力了，我们考虑既然我们已经维护好了之前的剖分信息，我们为什么不一步直接跳到当前链的顶端，然后跳到他的father，再到顶端再到father，用这样类似于树剖的方式来简化复杂度呢？
答案是可以的,我们引入splay来维护这些重链(也就是说一个重链对应了一个splay),splay的根的父亲为这个重链顶端的父亲(树根所在重链的splay的根为0),一个点左子树的点深度比当前点小,右子树的点深度比当前点大。这样我们就可以把原图等价的转化成若干个splay,然后这些splay通过father来”挂钩”起来的效果。那么我们跳到当前重链的顶端实际上就是寻找splay的最左边的端点,但我们注意到,我们splay维护的重链在父子关系上是没错的,splay的确能还原出那条重链，但是最左边端点的父亲并不是这个重链的父亲,而是这个重链的第二个点。这个重链的父亲是根的父亲(之前已经交代过了)。所以我们只需要把被访问的点splay到根，他的父亲对应的就是重链的父亲。
我们解决完了爬树的问题,但我们该怎么维护树链上的信息呢？根据我们的剖分方式，被访问的点会把他到树根直接连成一条重链,那我们就会破坏掉某些重链。那么这些重链的splay也会变。我们考虑这个过程中哪里重链的信息会被改变。显然是我们走到一个重链的顶端,然后跳他的father的时候，因为他的father所在的重链要和当前重链合并起来,那么他的father原本延伸下去的重链就要被断开。那么这个操作也非常的easy,大家可以回想一下splay是怎么维护重链的。我们只需要把father splay到根，那么他的右子树就是他所延伸下去的重链。我们把他的右子树直接断开(我们可以注意到他的父子信息仍然是对的,延伸下去的重链的father指向了father)。然后把右子树接上重链的顶端。那么2个重链就合并完成了。
值得一提的是这部分就是整个lct的核心操作access，就这么被讲完了。是不是感觉lct很easy?但是我们只是维护好了lct的剖分方式,并没有涉及到他的核心应用。我们接下来思考要怎么加边(link),删边(cut)。
既然谈到了加边删边,那么我们就必须说明图是不连通的了,这张图是一个森林。对于加边操作,我一开始的想法是fa[x]=y。但这只有y是一个点的时候才正确,因为一条边x,y是能把x的子树和y的子树给并起来的。把2个树并到一起就非常的麻烦了,尤其是我们还是用splay来维护链的时候。容易发现两个树的时候,其实两个树完全可以独立开看,就当于是2个之前我们描述的结构。假设我们合并完这2个树后,树是以y树为主体的，即x是挂在y的子树里的。那如果说x是x树的树根的话，我们显然只要fa[x]=y即可。但是x现在不是x树的树根。
我们考虑引入一个操作make_root。这个操作是把一个点操作成他所在的树的树根。那实际上就是点的深度以及他的父子关系改变了,考虑到我们的splay维护的是深度序,splay和father都要相应的改变。假设我们选定的点是x,容易发现x的祖先中,除了x子树以外的子树的相对深度和father全部不变。但是各个祖先的深度和father发生了改变,且是翻转过来了。那么我们access(x),然后splay(x),容易发现这时候x所有的祖先都挂在x的左子树上面,我一开始的想法是交换x的左子树和右子树。后来我发现我想当然了,因为在splay中x的祖先显然不会是呈一条链的挂在左子树上,如果是这样要splay还有什么用,所以我们要传一个翻转标记下去。那为什么我们不用修改splay上的father呢？这时候splay的灵活性就表现出来了,因为splay维护重链的深度序的同时,已经灵活的把他的father表示出来了,而翻转后他的father正好是一一对应上的。而且通过splay这个结构,我们可以很轻易的获得一些链上的信息。只需要make_root(y),access(x),然后访问splay维护出来的信息即可。
再回到link操作,显然只要make_root(x),fa[x]=y即可。
再看cut操作,考虑逆link操作。我们先make_root(x),access(y),注意这时候因为访问了y,x不一定在splay的根,再splay(x),因为y和x是直接有边相连的,所以y肯定在x右儿子,并且y下面是不挂东西的。然后我们fa[y]=rc[x]=0;即可
那么我们就讲完了link_cut_tree了,是不是感觉很easy呢

sol：

模板题

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdio>
#include<cstdlib>
using namespace std;

const int N=250000;

int n,m;
int q[N],lc[N],rc[N],fa[N];
int size[N],a[N];
bool rev[N];

int read()
{
    int res=0;
    char c;
    bool pd=false;
    while(((c=getchar())>'9'||c<'0')&&c!='-');
    if(c=='-')pd=true;
    else res=c-'0';
    while((c=getchar())>='0'&&c<='9')
    res=res*10+c-'0';
    if(pd)res=-res;
    return res;
}

inline int updata(const int &x)
{
    size[x]=size[lc[x]]+size[rc[x]]+1;
}

inline bool is_root(const int &x)
{
    return (lc[fa[x]]!=x&&rc[fa[x]]!=x);
}

inline void tag_rev(const int &x)
{
    rev[x]=!rev[x];
    swap(lc[x],rc[x]);
}
inline void tag_down(const int &x)
{
    if(rev[x])
    {
        tag_rev(lc[x]);
        tag_rev(rc[x]);
        rev[x]=!rev[x];
    }
}

inline void rotate(const int &x)
{
    int y=fa[x],z=fa[y],b;
    b=x==lc[y]?rc[x]:lc[x];
    fa[x]=z; fa[y]=x;
    if(b) fa[b]=y;
    if(z)
    {
        if(lc[z]==y) lc[z]=x;
        else if(rc[z]==y) rc[z]=x;
    }
    if(lc[y]==x) rc[x]=y,lc[y]=b;
    else lc[x]=y,rc[y]=b;
    updata(y);
}

inline void splay(const int &x)
{
    q[q[0]=1]=x;
    for(int i=x;!is_root(i);i=fa[i])
        q[++q[0]]=fa[i];
    while(q[0]>0)
        tag_down(q[q[0]--]);
    while(!is_root(x))
    {
        if(!is_root(fa[x]))
        if((lc[fa[x]]==x)==(lc[fa[fa[x]]]==fa[x]))
        rotate(fa[x]);
        rotate(x);
    }
    updata(x); 
}

inline void access(const int &x)
{
    for(int q=x,p=0;q;p=q,q=fa[q])
    {
        splay(q);
        rc[q]=p;
        updata(q);
    }
}

inline void make_root(const int &x)
{
    access(x);
    splay(x);
    tag_rev(x);
}

inline void link(const int &x,const int &y)
{
    make_root(x);
    fa[x]=y;
}

inline void cut(const int &x,const int &y)
{
    make_root(x);
    access(y);
    splay(x);
    fa[y]=0;rc[x]=0;
    updata(x);
}

int main()
{
//  freopen("bounce.in","r",stdin);
//  freopen("bounce.out","w",stdout);
    int i,j,k,flag;
    n=read();
    for(i=1;i<=n;++i)
    {
        a[i]=read();
        link(i,min(i+a[i],n+1));
//      link(min(i+a[i],n+1),i);
    }
    m=read();
    for(i=1;i<=m;++i)
    {
        flag=read();
        j=read()+1;
        if(flag-1)
        {
            cut(j,min(j+a[j],n+1));
//          cut(min(j+a[j],n+1),j);
            a[j]=read();
            link(j,min(j+a[j],n+1));
//          link(min(j+a[j],n+1),j);
        }
        else
        {
            make_root(n+1);
            access(j);
            splay(n+1);
            printf("%d\n",size[n+1]-1);
        }
    }
}