BZOJ_2002_[Hnoi2010]Bounce 弹飞绵羊_LCT
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
1 2 1 1
3
1 1
2 1 1
1 1
Sample Output
2
3
每个位置i向i+ki的位置连一条有向边,如果出去了就向n+1连。
形成了一棵树,每次询问相当于求一个点的深度。
这里直接当成无根树来做,每次询问要把n+1变成根。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 200050
#define ls ch[p][0]
#define rs ch[p][1]
#define get(x) (ch[f[x]][1]==x)
int a[N],n,f[N],ch[N][2],rev[N],siz[N];
bool isrt(int x) {return ch[f[x]][1]!=x&&ch[f[x]][0]!=x;}
void pushup(int p) {
if(!p) return ;
siz[p]=1;
if(ls) siz[p]+=siz[ls];
if(rs) siz[p]+=siz[rs];
}
void pushdown(int p) {
if(rev[p]) {
rev[ls]^=1; swap(ch[ls][0],ch[ls][1]);
rev[rs]^=1; swap(ch[rs][0],ch[rs][1]);
rev[p]=0;
}
}
void update(int p) {
if(!isrt(p)) update(f[p]);
pushdown(p);
}
void rotate(int x) {
int y=f[x],z=f[y],k=get(x);
if(!isrt(y)) ch[z][ch[z][1]==y]=x;
ch[y][k]=ch[x][!k]; f[ch[y][k]]=y;
ch[x][!k]=y; f[y]=x; f[x]=z;
pushup(y); pushup(x);
}
void splay(int x) {
update(x);
for(int d;d=f[x],!isrt(x);rotate(x))
if(!isrt(d))
rotate(get(x)==get(d)?d:x);
}
void access(int p) {
for(int t=0;p;t=p,p=f[p])
splay(p),rs=t,pushup(p);
}
void makeroot(int p) {
access(p); splay(p); rev[p]^=1; swap(ls,rs);
}
void link(int x,int p) {
makeroot(x); splay(p); f[x]=p;
}
void cut(int x,int p) {
makeroot(x); access(p); splay(p); ls=f[x]=0;
}
int main() {
scanf("%d",&n);
int i;
for(i=1;i<=n+1;i++) siz[i]=1;
for(i=1;i<=n;i++) {
scanf("%d",&a[i]);
link(i,min(n+1,i+a[i]));
}
int m,opt,x,y;
scanf("%d",&m);
while(m--) {
scanf("%d%d",&opt,&x); x++;
if(opt==1) {
makeroot(n+1); access(x); splay(x); printf("%d\n",siz[ch[x][0]]);
}else {
scanf("%d",&y);
cut(x,min(n+1,x+a[x])); a[x]=y; link(x,min(n+1,x+a[x]));
}
}
}