BZOJ 2002 弹飞绵羊（分块）

题目：弹飞绵羊

这道题，据说是lct裸题，但是lct那么高级的数据结构，我并不会，所以采取了学长讲过的分块做法，我们对序列分块，可以定义两个数组，其中一个表示从当前位置跳出当前块需要多少步，另一个数组表示从当前位置跳到下一块会落在哪个位置，然后新修改就暴力修改当前块，查询就直接暴力跑块外的结果。数组初始化可以考虑倒着跑，然后分情况讨论。这题被我们完美解决了。

 

下面上代码：

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define db double
using namespace std;
const int N=200005;
int n,m,a[N],t1,t2,t3,siz,s,t,to[N],stp[N];
int query(int p){
    int rt=0;while(~p){
        rt+=stp[p];p=to[p];
    } return rt;
} void update(int p,int da){
    s=p/siz*siz;t=(p/siz+1)*siz-1;
    a[p]=da;for(int i=p;i>=s;i--)
    if(i+a[i]>=n) to[i]=-1,stp[i]=1;
    else if(i+a[i]>t) to[i]=i+a[i],stp[i]=1;
    else to[i]=to[i+a[i]],stp[i]=stp[i+a[i]]+1;
} int main(){
    scanf("%d",&n);siz=(int)sqrt((db)n+0.5);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int i=n-1;~i;i--){
        t=(i/siz+1)*siz-1;
        if(i+a[i]>=n) to[i]=-1,stp[i]=1;
        else if(i+a[i]>t) 
        to[i]=i+a[i],stp[i]=1;
        else to[i]=to[i+a[i]],
        stp[i]=stp[i+a[i]]+1;
    } scanf("%d",&m);
    while(m--){
        scanf("%d%d",&t1,&t2);
        if(t1==1) printf("%d\n",query(t2));
        else scanf("%d",&t3),update(t2,t3);
    } return 0;
}