CF797E Array Queries

written on 2022-05-25

根号分治题，其实是P3203弹飞绵羊的另一种版本，本质是相同的。

我想了想，两者的主要区别在于那题维护的是跳出本块，这题维护的是跳完的次数，那么这个区别的主要原因其实在于一个带修一个不带修。如果带修的话，那么维护的时候就还要维护前面的信息，这样的话为了不让时间复杂度退化，就选择前一种维护方式，这样每次都只用对块内元素进行修改。

#include<bits/stdc++.h>
#define N 100005
using namespace std;
int n,B,num,id[N],a[N];
int L[320],R[320],cnt[N][320];
int main()
{
	scanf("%d",&n);
	B=sqrt(n);
	num=n/B;if(n%B) num++;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),id[i]=(i-1)/B+1;
	for(int i=1;i<=num;i++) L[i]=(i-1)*B+1,R[i]=i*B;
	R[num]=n;
	for(int i=n;i;i--)
	{
		for(int j=1;j<=B;j++)
		{
			if(i+a[i]+j>n) cnt[i][j]=1;
			else cnt[i][j]=cnt[i+a[i]+j][j]+1;
		}
	}
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int x,k;
		scanf("%d%d",&x,&k);
		if(k<=B) printf("%d\n",cnt[x][k]);
		else
		{
			int res=0;
			while(x<=n) x=x+k+a[x],res++;
			printf("%d\n",res);
		}
	}
}