我们把x向跳到的位置建边(跳出去了就是n+1),可以发现一定是一棵树。那么x跳的次数就是x到n+1之间的边数。每次修改就先把原来的断开,连上新的,用lct维护即可。哈哈哈改对了开心~
update:也可以暴力分块搞。
lct
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n,m,fa[N],c[N][2],sz[N],a[N],q[N];
bool rev[N];
inline bool isroot(int x){
return x!=c[fa[x]][0]&&x!=c[fa[x]][1];
}
inline void update(int x){
int l=c[x][0],r=c[x][1];
sz[x]=sz[l]+sz[r]+1;
}
inline void pushdown(int p){
if(!rev[p]) return;rev[p]=0;
int l=c[p][0],r=c[p][1];
rev[l]^=1;rev[r]^=1;swap(c[p][0],c[p][1]);
}
inline void rotate(int x){
int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
if(!isroot(y)) c[z][y==c[z][1]]=x;
fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x){
int top=0;q[++top]=x;
for(int xx=x;!isroot(xx);xx=fa[xx]) q[++top]=fa[xx];
while(top) pushdown(q[top--]);
while(!isroot(x)){
int y=fa[x],z=fa[y];
if(!isroot(y)){
if(x==c[y][1]^y==c[z][1]) rotate(y);
else rotate(x);
}rotate(x);
}
}
inline void access(int x){
int y=0;
while(x){splay(x);c[x][1]=y;update(x);y=x;x=fa[x];}
}
inline void makeroot(int x){
access(x);splay(x);rev[x]^=1;
}
inline void link(int x,int y){
makeroot(x);fa[x]=y;
}
inline void cut(int x,int y){
makeroot(x);access(y);splay(y);c[y][0]=fa[x]=0;update(y);
}
inline int solve(int x){
makeroot(n+1);
access(x);splay(x);
return sz[x];
}
int main(){
// freopen("a.in","r",stdin);
n=read();
for(int i=1;i<=n;++i){
a[i]=min(i+read(),n+1);link(i,a[i]);
}m=read();
while(m--){
int op=read(),x=read()+1;
if(op==1){printf("%d\n",solve(x)-1);continue;}
cut(x,a[x]);a[x]=min(x+read(),n+1);link(x,a[x]);
}return 0;
}
分块暴力
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
#define N 200010
#define ll long long
#define inf 0x3f3f3f3f
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n,m,a[N],id[N],nxt[N],cnt[N],nn,ans;
void change(int k){
int mx=k*nn-1,mn=(k-1)*nn;
if(k==(n-1)/nn+1) mx=n-1;
for(int i=mx;i>=mn;i--){
nxt[i]=a[i];cnt[i]=1;
if(nxt[i]<=mx) cnt[i]+=cnt[nxt[i]],nxt[i]=nxt[nxt[i]];
if(nxt[i]>=n) nxt[i]=-1;
}
}
void query(int i){
if(nxt[i]==-1) return;
ans+=cnt[nxt[i]];
query(nxt[i]);
}
int main(){
// freopen("a.in","r",stdin);
n=read();nn=sqrt(n);if(nn*nn<n) ++nn;
for(int i=0;i<n;++i) a[i]=i+read(),id[i]=i/nn+1;
for(int i=1;i<=(n-1)/nn+1;++i) change(i);m=read();
while(m--){
int op=read(),x=read();
if(op==1){
ans=cnt[x];query(x);printf("%d\n",ans);
}else a[x]=x+read(),change(id[x]);
}
return 0;
}