【bzoj2002】弹飞绵羊（分块）

题目分析

　　题意：每个点都有一个值$v_i$，从一个点出发，每走到一个点，会跳到i+vi的位置，问需要跳多少次能跳出n？带修改。

　　此题可以用lct做，此处使用了分块：将序列分块后，每个点记录从此点最少跳几次能跳出当前块，和跳出后到达的位置，倒叙可以优化。

　　这样询问时只要一直跳至多$\sqrt{n}$个块就能知道答案。

　　对于修改操作，只需将修改点和修改点之前的当前块元素重新使用上述处理，同样倒叙可以优化。

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;

const int N = 2e5 + ;
int n, m, S;
int blkCnt, blk[N], bl[], br[];
long long val[N], step[N], des[N];

inline int read(){
    int i = , f = ; char ch = getchar();
    for(; (ch < '' || ch > '') && ch != '-'; ch = getchar());
    if(ch == '-') f = -, ch = getchar();
    for(; ch >= '' && ch <= ''; ch = getchar())
        i = (i << ) + (i << ) + (ch - '');
    return i * f;
}

inline long long readL(){
    long long i = , f = ; char ch = getchar();
    for(; (ch < '' || ch > '') && ch != '-'; ch = getchar());
    if(ch == '-') f = -, ch = getchar();
    for(; ch >= '' && ch <= ''; ch = getchar())
        i = (i << ) + (i << ) + (ch - '');
    return i * f;
}

inline void wr(long long x){
    if(x < ) putchar('-'), x = -x;
    if(x > ) wr(x / );
    putchar(x %  + '');
}

inline void init(){
    bl[blkCnt = ] = ;
    for(int i = ; i <= n; i++){
        if(i % S == ){
            br[blkCnt] = i;
            blk[i] = blkCnt;
            if(i +  <= n){
                bl[++blkCnt] = i + ;
                continue;
            }
        }
        blk[i] = blkCnt;
    }
    br[blkCnt] = n;
}

int main(){
    n = read(), S = sqrt(n);
    for(int i = ; i <= n; i++) val[i] = readL();
    init();
    for(int i = n; i >= ; i--){
        int tmp = i;
        bool flag = false;
        step[i] = ;
        while(tmp + val[tmp] <= br[blk[i]]){
            tmp += val[tmp], step[i]++;
            if(step[tmp]){
                step[i] += step[tmp], des[i] = des[tmp], flag = true;
                break;
            }
        }
        if(!flag) des[i] = tmp + val[tmp], step[i]++;
    }
    m = read();
    for(int t = ; t <= m; t++){
        int opt = read();
        if(opt == ){
            int src = read() + , ans = , tmp;
            tmp = src;
            while(des[tmp] <= n) ans += step[tmp], tmp = des[tmp];
            ans += step[tmp];
            wr(ans), putchar('\n');
        }
        else{
            int pos = read() + , value = readL();
            if(val[pos] == value) continue;
            val[pos] = value;
            for(int i = bl[blk[pos]]; i <= pos; i++) step[i] = ;

            for(int i = pos; i >= bl[blk[pos]]; i--){
                int tmp = i; bool flag = false;
                while(tmp + val[tmp] <= br[blk[pos]]){
                    tmp += val[tmp], step[i]++;
                    if(step[tmp]){
                        step[i] += step[tmp], des[i] = des[tmp], flag = true;
                        break;
                    }
                }
                if(!flag) des[i] = tmp + val[tmp], step[i]++;
            }
        }
    }
    return ;
}