bzoj2002 [Hnoi2010]Bounce 弹飞绵羊

Description

---

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

20%的数据n,m<=10000
100%的数据n<=200000,m<=100000

Solution

---

又是一道LCT裸题。看完题面就脑补出了一棵动来动去的树-_-||
这里要求点到根的距离，显然不能暴力跳。splay维护一个区间的大小，先access(x)再把x旋到splay的根就可以知道通过根左儿子的size知道深度了
据说可以分块做

Code

---

#include <stdio.h>
#include <algorithm>
const int N=400005;
struct treeNode{int son[2],fa,lazy,size; bool is_root;}t[N];
int a[N],n;
void push_up(int x) {
    t[x].size=t[t[x].son[0]].size+t[t[x].son[1]].size+1;
}
void push_down(int x) {
    if (!t[x].lazy||!x) return ;
    std:: swap(t[x].son[0],t[x].son[1]);
    t[t[x].son[0]].lazy^=1;
    t[t[x].son[1]].lazy^=1;
    t[x].lazy=0;
}
void rotate(int x) {
    if (t[x].is_root) return ;
    int y=t[x].fa; int z=t[y].fa;
    int k=t[y].son[1]==x;
    t[y].son[k]=t[x].son[!k];
    if (t[x].son[!k]) t[t[x].son[!k]].fa=y;
    t[x].son[!k]=y;
    t[y].fa=x;
    t[x].fa=z;
    if (t[y].is_root) {
        t[x].is_root=1;
        t[y].is_root=0;
    } else t[z].son[t[z].son[1]==y]=x;
    push_up(y);
    push_up(x);
}
void remove(int x) {
    if (!t[x].is_root) remove(t[x].fa);
    push_down(x);
}
void splay(int x) {
    remove(x);
    while (!t[x].is_root) {
        int y=t[x].fa; int z=t[y].fa;
        if (!t[y].is_root) {
            if ((t[y].son[1]==x)^(t[z].son[1]==y)) rotate(x);
            else rotate(y);
        }
        rotate(x);
    }
}
void access(int x) {
    int y=0;
    do {
        splay(x);
        t[t[x].son[1]].is_root=1;
        t[t[x].son[1]=y].is_root=0;
        push_up(x);
        y=x; x=t[x].fa;
    } while (x);
}
void mroot(int x) {
    access(x); splay(x);
    t[x].lazy^=1;
}
void link(int x,int y) {
    if (y>n) y=n+1;
    mroot(x); t[x].fa=y;
    splay(x);
}
void cut(int x,int y) {
    if (y>n) y=n+1;
    mroot(x); access(y); splay(y);
    t[t[y].son[0]].is_root=1;
    t[t[y].son[0]].fa=0;
    t[y].son[0]=0;
    push_up(y);
}
int query(int x) {
    mroot(n+1);
    access(x); splay(x);
    return t[t[x].son[0]].size;
}
int main(void) {
    scanf("%d",&n);
    for (int i=1;i<=n+1;i++) t[i].is_root=t[i].size=1;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=n;i++) {
        if (i+a[i]>n) link(i,n+1);
        else link(i,i+a[i]);
    }
    int m; scanf("%d",&m);
    while (m--) {
        int opt,x; scanf("%d%d",&opt,&x); x++;
        if (opt==1) {
            printf("%d\n", query(x));
        } else {
            int y; scanf("%d",&y);
            cut(x,x+a[x]);
            a[x]=y;
            link(x,x+a[x]);
        }
    }
    return 0;
}