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;}