50PTS分块求助

#include <bits/stdc++.h>
#define prs " \n"
#define LINF 0x3F3F3F3F3F3F3F3FLL
using namespace std;
using i64=long long;
using u64=unsigned long long;
const int N=2E5+10;
int n,a[N],m,bl;
int tb[N],reb[N];
int getbid(int o)
{
	if(o>n) return N;
	return ((o-1)/bl)+1;
}
int quer(int o)
{
	int res=0;
	while(o<=n)
	{
		res+=reb[o];
		o=tb[o];
	}
	return res;
}
void modif(int o)
{
	int k,j=o,sts=0;
	cin>>k;
	a[o]=k;
	while(getbid(o)==getbid(j)) 
		j+=a[j],sts++;
	tb[o]=j,reb[o]=sts;
	for(int i=(getbid(o)-1)*bl+1;i<o;i++)
		if(i+a[i]==o) 
			tb[i]=tb[o],reb[i]=reb[o]+1;
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n;
	bl=sqrt(n);
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
	{
		int j=i,sts=0;
		while(getbid(i)==getbid(j)) j+=a[j],sts++;
		tb[i]=j,reb[i]=sts;
	}
	cin>>m;
	while(m--)
	{
		int o,k;
		cin>>o>>k;
		k++;
		if(o==1) cout<<quer(k)<<'\n';
		else modif(k);
	}
	return 0;
}