AC日记——[HNOI2010]BOUNCE 弹飞绵羊 洛谷 P3203

[HNOI2010]BOUNCE 弹飞绵羊

思路：

　　SBlct；

代码：

#include <bits/stdc++.h>
using namespace std;
#define maxn 200005
int n,m,f[maxn],ch[maxn][],rev[maxn],ki[maxn],sta[maxn],top,lit,size[maxn];
inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'')Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
void updata(int now)
{
    size[now]=size[ch[now][]]+size[ch[now][]]+;
}
void downdata(int now)
{
    if(rev[now])
    {
        swap(ch[now][],ch[now][]);
        rev[now]^=,rev[ch[now][]]^=,rev[ch[now][]]^=;
    }
}
bool isroot(int now)
{
    return (ch[f[now]][]!=now)&&(ch[f[now]][]!=now);
}
void rotate(int now)
{
    int fa=f[now],ffa=f[fa],l=(ch[fa][]==now),r=l^;
    if(!isroot(fa)) ch[ffa][ch[ffa][]==fa]=now;
    f[now]=ffa,f[fa]=now,ch[fa][l]=ch[now][r],ch[now][r]=fa;
    if(ch[fa][l]) f[ch[fa][l]]=fa;
    updata(fa),updata(now);
}
void splay(int now)
{
    top=,sta[]=now;
    for(int i=now;!isroot(i);i=f[i]) sta[++top]=f[i];
    while(top) downdata(sta[top--]);
    int fa,ffa;
    while(!isroot(now))
    {
        fa=f[now],ffa=f[fa];
        if(!isroot(fa))
        {
            if((ch[fa][]==now)^(ch[ffa][]==fa)) rotate(fa);
            else rotate(now);
        }
        rotate(now);
    }
}
void access(int now)
{
    for(int i=;now;i=now,now=f[now]) splay(now),ch[now][]=i;
}
void makeroot(int now)
{
    access(now),splay(now),rev[now]^=;
}
void split(int x,int y)
{
    makeroot(x),access(y),splay(y);
}
void cut(int x,int y)
{
    split(x,y),f[x]=ch[y][]=;
}
void link(int x,int y)
{
    makeroot(x),f[x]=y;
}
int ofans(int x)
{
    makeroot(lit),access(x),splay(x);
    return size[ch[x][]];
}
int main()
{
    freopen("data.txt","r",stdin);
    in(n),lit=n+;int u,v,op;
    for(int i=;i<=n;i++)
    {
        in(ki[i]);
        if(i+ki[i]<=n) link(i,i+ki[i]);
        else link(i,lit);
    }
    in(m);
    while(m--)
    {
        in(op);
        if(op==) in(u),u++,printf("%d\n",ofans(u));
        else
        {
            in(u),in(v),u++;
            if(u+ki[u]<=n) cut(u,u+ki[u]);
            else cut(u,lit);
            ki[u]=v;
            if(u+v<=n) link(u,u+v);
            else link(u,lit);
        }
    }
    return ;
}