BZOJ 2002: [Hnoi2010]Bounce 弹飞绵羊 LCT

来源:互联网 发布:picasa3 mac 编辑:程序博客网 时间:2024/03/29 04:37


LCT 维护子树大小.....

2002: [Hnoi2010]Bounce 弹飞绵羊

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 4085  Solved: 2188
[Submit][Status]

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

Splay 启发式合并

[Submit][Status]




#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int maxn=220000;int n,m;int pre[maxn],ch[maxn][2],size[maxn];bool rt[maxn];void init(){  for(int i=0;i<=n+10;i++)    rt[i]=1,pre[i]=0,ch[i][0]=ch[i][1]=0,size[i]=1;  size[0]=0;}void push_up(int r){  size[r]=size[ch[r][0]]+size[ch[r][1]]+1;}void Rotate(int x){  int y=pre[x],kind=ch[y][1]==x;  ch[y][kind]=ch[x][!kind];  pre[ch[y][kind]]=y;  pre[x]=pre[y];  pre[y]=x;  ch[x][!kind]=y;  if(rt[y]) rt[y]=false,rt[x]=true;  else ch[pre[x]][ch[pre[x]][1]==y]=x;  push_up(y);}void Splay(int r){  while(!rt[r])    {      int f=pre[r],ff=pre[f];      if(rt[f]) Rotate(r);      else if((ch[ff][1]==f)==(ch[f][1]==r)) Rotate(f),Rotate(r);      else Rotate(r),Rotate(r);    }  push_up(r);}int Access(int x){  int y=0;  for(;x;x=pre[y=x])    {      Splay(x);      rt[ch[x][1]]=true; rt[ch[x][1]=y]=false;      push_up(x);    }  return y;}int main(){  while(scanf("%d",&n)!=EOF)    {      init();      for(int i=1;i<=n;i++)        {          int t;          scanf("%d",&t);          pre[i]=(i+t>n+1)?n+1:i+t;        }      scanf("%d",&m);      while(m--)        {          int tk,u,v;          scanf("%d",&tk);          if(tk==1)            {              scanf("%d",&u);              u++;              Access(u);              Splay(u);              printf("%d\n",size[ch[u][0]]);            }          else            {              scanf("%d%d",&u,&v);              u++;              Access(u);              Splay(u);              pre[ch[u][0]]=pre[u];              pre[u]=0;              rt[ch[u][0]]=true;              ch[u][0]=0;              push_up(u);              pre[u]=(u+v)>n+1?n+1:u+v;            }        }    }  return 0;}


1 0
原创粉丝点击