【bzoj 2002】弹飞绵羊

时间:2023-03-09 01:56:01
【bzoj 2002】弹飞绵羊

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

唉隔壁都在弹飞大爷了,我还在弹绵羊>_<

分块大法【Time:1472 ms】

 #include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=;
int n,m,num,pl,w,bl,cnt;
int k[N],be[N],st[N],to[N];
int read()
{
int x=,f=;char c=getchar();
while(c<''||c>''){if(c=='-')f=-;c=getchar();}
while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
return x*f;
}
int ask(int t)
{
int ans=st[t];
while(to[t]<=n)
{
t=to[t];
ans+=st[t];
}
return ans;
}
int main()
{
n=read();bl=sqrt(n);
if(n%bl)cnt=n/bl+;
else cnt=n/bl;
for(int i=;i<=n;i++)
{
k[i]=read();
be[i]=(i-)/bl+;
}
for(int i=n;i>;i--)
{
if(be[i]==be[i+k[i]]){st[i]=st[i+k[i]]+;to[i]=to[i+k[i]];}
else {st[i]=;to[i]=i+k[i];}
}
m=read();
while(m--)
{
num=read();pl=read()+;
if(num==)printf("%d\n",ask(pl));
else
{
w=read();
k[pl]=w;
int l=(be[pl]-)*bl+;
for(int i=pl;i>=l;i--)
{
if(be[i]==be[i+k[i]]){st[i]=st[i+k[i]]+;to[i]=to[i+k[i]];}
else {st[i]=;to[i]=i+k[i];}
}
}
}
return ;
}

LCT【Time:1844 ms】

 #include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=;
int n,m,c[N][],next[N],fa[N],size[N],st[N];
bool rev[N];
int read()
{
int x=,f=;char c=getchar();
while(c<''||c>''){if(c=='-')f=-;c=getchar();}
while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
return x*f;
}
bool isroot(int k){return c[fa[k]][]!=k&&c[fa[k]][]!=k;}
void up(int x){size[x]=size[c[x][]]+size[c[x][]]+;}
void down(int x)
{
int l=c[x][],r=c[x][];
if(rev[x]){rev[x]^=;rev[l]^=;rev[r]^=;swap(c[x][],c[x][]);}
}
void rotate(int x)
{
int y=fa[x],z=fa[y],l,r;
if(c[y][]==x)l=;else l=;r=l^;
if(!isroot(y)){if(c[z][]==y)c[z][]=x;else c[z][]=x;}
fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
c[y][l]=c[x][r];c[x][r]=y;
up(y);up(x);
}
void splay(int x)
{
int top=;st[++top]=x;
for(int i=x;!isroot(i);i=fa[i])st[++top]=fa[i];
for(int i=top;i;i--)down(st[i]);
while(!isroot(x))
{
int y=fa[x],z=fa[y];
if(!isroot(y))
{
if((c[y][]==x)^(c[z][]==y))rotate(x);
else rotate(y);
}
rotate(x);
}
}
void acs(int x)
{
int t=;
while(x){splay(x);c[x][]=t;t=x;x=fa[x];}
}
void mkroot(int x)
{
acs(x);splay(x);rev[x]^=;
}
void link(int x,int y)
{
mkroot(x);fa[x]=y;splay(x);
}
void cut(int x,int y)
{
mkroot(x);acs(y);splay(y);c[y][]=fa[x]=;
}
int main()
{
n=read();
for(int i=;i<=n;i++)
{
int x=read();
fa[i]=x+i;size[i]=;
if(fa[i]>n+)fa[i]=n+;
next[i]=fa[i];
}
size[n+]=;
m=read();
while(m--)
{
int f=read();
if(f==)
{
mkroot(n+);
int x=read()+;
acs(x);splay(x);printf("%d\n",size[c[x][]]);
}
else
{
int x=read()+,y=read();
int t=min(n+,x+y);
cut(x,next[x]);link(x,t);next[x]=t;
}
}
return ;
}