[HNOI2010]弹飞绵羊

题目大意:
  给你一个长度为$n(n\leq200000)$的正整数数列$k$。数列中按照如下方式进行位置跳转,若当前所在的位置为$i$,则下一步跳转到$i+k_i$的位置上。要求进行如下两种操作:
  1.将$k_x$的值修改为$y$;
  2.询问从$k_x$出发,需要几步跳出整个数列?

思路:
  分块,记录每个位置跳出当前块所需的步数和跳出块后所到达的位置。每次修改暴力修改整个块,复杂度$O(\sqrt n)$。询问时暴力往后跳,复杂度$O(\sqrt n)$。总时间复杂度$O(n+m\sqrt n)$。

 1 #include<cmath>
 2 #include<cstdio>
 3 #include<cctype>
 4 inline int getint() {
 5     register char ch;
 6     while(!isdigit(ch=getchar()));
 7     register int x=ch^'0';
 8     while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
 9     return x;
10 }
11 const int N=200000;
12 int n,k[N],bel[N],begin[N],end[N],step[N],to[N];
13 inline int query(int x) {
14     int ret=0;
15     while(x<n) {
16         ret+=step[x];
17         x=to[x];
18     }
19     return ret;
20 }
21 inline void update(const int &b,const int &e) {
22     for(register int i=e;i>=b;i--) {
23         if(i+k[i]>end[bel[i]]) {
24             step[i]=1;
25             to[i]=i+k[i];
26         } else {
27             step[i]=step[i+k[i]]+1;
28             to[i]=to[i+k[i]];
29         }
30     }
31 }
32 int main() {
33     const int block=sqrt(n=getint());
34     for(register int i=0;i<n;i++) {
35         k[i]=getint();
36         bel[i]=i/block;
37         if(bel[i]&&!begin[bel[i]]) begin[bel[i]]=i;
38         end[bel[i]]=i;
39     }
40     update(0,n-1);
41     for(register int i=getint();i;i--) {
42         const int opt=getint(),x=getint();
43         if(opt==1) {
44             printf("%d\n",query(x));
45         } else {
46             k[x]=getint();
47             update(begin[bel[x]],x);
48         }
49     }
50     return 0;
51 }

 

posted @ 2018-02-21 17:47  skylee03  阅读(108)  评论(0编辑  收藏  举报