Codeforces 797E Array Queries 分块思想

点击打开链接

题意:给出n个数 ai<=n,q次询问,n,q<=1e5,每次询问给出p,k,每次op将p->p+a[p]+k 问经过多少次p才能>n?

p每次都加上固定的k和变化的a[p],当k>sqrt(n) 则最多循环sqrt(n)次,则可以预处理出k<=sqrt(n)的询问,时间复杂度O(Q*sqrt(N)) 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+20;
int a[N],n,q;
int t[500][N];//t[k][p] 
int main()
{
	while(cin>>n)
	{
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		int b=sqrt(n);
		memset(t,0,sizeof(t));
		for(int k=1;k<=b;k++)
		{
			for(int i=n;i>=1;i--)
			{
				t[k][i]=t[k][i+a[i]+k]+1;
			}
		}
		cin>>q;
		int p,k;
		while(q--)
		{
			int cnt=0;
			scanf("%d%d",&p,&k);
			if(k<=b)
				printf("%d\n",t[k][p]);
			else
			{
				while(p<=n)
				{
					p=p+a[p]+k;
					cnt++;
				}
				printf("%d\n",cnt);
			}
		}	
	} 
	return 0;
}