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
题解:动态树。
将每个点的父亲置为能弹到的点,注意是单向边,link()时不用makeroot()。
询问相当于求深度,直接access(),splay()输出size即可。
代码:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
const int Maxn=2e5+50;
inline int read()
{
char ch=getchar();int i=0,f=1;
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){i=(i<<1)+(i<<3)+ch-'0';ch=getchar();}
return i*f;
}
int n,Q,k[Maxn];
struct node
{
node *lc,*rc,*fa;
int sze;
node():lc(NULL),rc(NULL),fa(NULL),sze(1){}
inline void upt()
{
sze=((lc?lc->sze:0)+(rc?rc->sze:0)+1);
}
}*tr[Maxn],Pool[Maxn],*pool=Pool;
inline bool isroot(node *x)
{
return !x->fa||(x->fa->lc!=x&&x->fa->rc!=x);
}
inline void rotate(node *x)
{
node *y=x->fa,*z=y->fa;
if(z&&!isroot(y))
{
if(z->lc==y)z->lc=x;
else z->rc=x;
}
if(x==y->lc)
{
node *b=x->rc;
if(b)b->fa=y;
x->rc=y;
y->lc=b;
}
else
{
node *b=x->lc;
if(b)b->fa=y;
x->lc=y;
y->rc=b;
}
x->fa=z,y->fa=x;
y->upt(),x->upt();
}
inline bool which(node *x)
{
return x->fa->lc==x;
}
inline void splay(node *x)
{
while(!isroot(x))
{
if(!isroot(x->fa))
{
if(which(x->fa)^which(x))rotate(x);
else rotate(x->fa);
}
rotate(x);
}
}
inline void access(node *x)
{
for(node *y=NULL;x;y=x,x=x->fa)
{
splay(x),x->rc=y;
if(y)y->fa=x;
x->upt();
}
}
inline void cut(node *x,node *y)
{
access(x);
splay(x);
x->lc->fa=NULL;
x->lc=NULL;
x->upt();
}
inline void link(node *x,node *y)
{
access(x);
splay(x);
x->fa=y;
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
{
tr[i]=++pool;
}
for(int i=1;i<=n;i++)
{
k[i]=read();
if(i+k[i]<=n)link(tr[i],tr[i+k[i]]);
else continue;
}
Q=read();
while(Q--)
{
int x=read();
if(x==1)
{
int y=read();
y++;
if(y>n)puts("0");
else
{
access(tr[y]);
splay(tr[y]);
printf("%d\n",tr[y]->sze);
}
}
else
{
int a=read(),b=read();
a++;
if(a>n)continue;
if(a+k[a]<=n)cut(tr[a],tr[a+k[a]]);
if(a+b<=n)link(tr[a],tr[a+b]);
k[a]=b;
}
}
}