Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 11244 Solved: 5719
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,如果可以跳到一个点,那就认爸爸
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=200000+10;
struct Tree{
int ch[2],size,fa;
}tree[MAXN];
int n,m,temp,kk,opt;
void pushup(int x){
tree[x].size=1;
if(tree[x].ch[0]) tree[x].size+=tree[tree[x].ch[0]].size;
if(tree[x].ch[1]) tree[x].size+=tree[tree[x].ch[1]].size;
}
bool isroot(int x){
return tree[tree[x].fa].ch[0]!=x&&tree[tree[x].fa].ch[1]!=x;
}
void rotate(int x){
int y=tree[x].fa,z=tree[y].fa,l,r;
if(tree[y].ch[0]==x)l=0;else l=1;r=l^1;
if(!isroot(y)){
if(tree[z].ch[0]==y)tree[z].ch[0]=x;
else tree[z].ch[1]=x;
}
tree[x].fa=z;tree[tree[x].ch[r]].fa=y;tree[y].fa=x;
tree[y].ch[l]=tree[x].ch[r];tree[x].ch[r]=y;
pushup(y);pushup(x);
}
void splay(int x){
while(!isroot(x)){
int y=tree[x].fa,z=tree[y].fa;
if(!isroot(y)){
if(tree[y].ch[0]==x^tree[z].ch[0]==y)rotate(x);
else rotate(y);
}
rotate(x);
}
}
void access(int x){for(register int i=0;x;i=x,x=tree[x].fa){splay(x);tree[x].ch[1]=i;pushup(x);}}
void link(int u,int v){
if(v>n)v=0;
access(u);
splay(u);
tree[tree[u].ch[0]].fa=0;
tree[u].ch[0]=0;
tree[u].fa=v;
pushup(u);
}
int main(){
scanf("%d",&n);
for(register int i=1;i<=n;i++){
tree[i].size = 1;
tree[i].fa=tree[i].ch[0]=tree[i].ch[1]=0;
scanf("%d",&temp);
link(i,i+temp);
}
tree[0].size=0;
scanf("%d",&m);
for(register int i=1;i<=m;i++){
scanf("%d",&opt);
if(opt==1){
scanf("%d",&temp);
temp++;
access(temp);splay(temp);
printf("%d\n",tree[tree[temp].ch[0]].size+1);
}else{
scanf("%d",&temp);temp++;scanf("%d",&kk);
link(temp,temp+kk);
}
}
return 0;
}