LCT维护子树大小就可以了。
每个点i跳到的位置为min(x+i,n+1)。
改变的时候,先和原先的a[x]断开,将a[x]至为当前更改后可跳到的位置,并和x连接。(a[x]没更新。。蠢得没法)查询答案就是点x到n+1之间点的个数-1。因为最后跳到n+1后就没有了,很好想。
以前用分块写过。。好像bzoj上测得差不多。。
【代码】
#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 200005
#define M 2001005
#define INF 1e9
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;
int read()
{
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return x*f;
}
int n,m,top,cnt;
int son[N][2],st[N],fa[N],sum[N],a[N];
bool rev[N];
void pushup(int p){
sum[p]=sum[son[p][0]]+sum[son[p][1]]+1;
}
bool isroot(int x){
return son[fa[x]][0]!=x&&son[fa[x]][1]!=x;
}
void pushdown(int k)
{
if(!rev[k]) return;
int l=son[k][0],r=son[k][1];
rev[k]=0,rev[l]^=1;rev[r]^=1;
swap(son[k][0],son[k][1]);
}
void Rotate(int x)
{
int y=fa[x],z=fa[y],l,r;
l=son[y][1]==x;r=l^1;
if(!isroot(y)) son[z][son[z][1]==y]=x;
fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
son[y][l]=son[x][r],son[x][r]=y;
pushup(y);pushup(x);
}
void Splay(int x)
{
st[++top]=x;
for(int i=x;!isroot(i);i=fa[i]) st[++top]=fa[i];
while(top) pushdown(st[top--]);
while(!isroot(x))
{
int y=fa[x],z=fa[y];
if(!isroot(y)) {
if(son[z][0]==y^son[y][0]==x) Rotate(x);
else Rotate(y);
}
Rotate(x);
}
}
void Access(int x){
for(int t=0;x;t=x,x=fa[x])
Splay(x),son[x][1]=t,pushup(x);
}
void Make_Root(int x){
Access(x);Splay(x);rev[x]^=1;
}
void Link(int x,int y){
Make_Root(x);fa[x]=y;
}
void Cut(int x,int y){
Make_Root(x);Access(y);Splay(y);
son[y][0]=fa[x]=0;pushup(y);
}
void Solve_Change(int x,int y)
{
y=min(x+y,n+1);
if(y==a[x]) return;
Cut(x,a[x]);Link(x,y);a[x]=y;
}
void Solve_Query(int x)
{
Make_Root(n+1);Access(x);Splay(x);
printf("%d\n",sum[x]-1);
}
void Input_Init()
{
n=read();
for(int i=1;i<=n+1;i++) sum[i]=1;
for(int i=1;i<=n;i++)
{
a[i]=min(read()+i,n+1);
Link(i,a[i]);
}
}
void Solve()
{
m=read();
while(m--)
{
static int f,x;
f=read(),x=read()+1;
if(f&1) Solve_Query(x);
else Solve_Change(x,read());
}
}
int main()
{
Input_Init();
Solve();
return 0;
}