[Codeforces 13E] Holes （分块）

Codeforces - 13E

有 N 个洞排成一列，每个洞有一个能量值 ai
将一个球丢进第 i 个洞内，它将跳跃到 i+ai 个洞
这样一直跳跃直到跳出 N 的范围
现在有两种操作，往第 i 个洞内丢球，将第 i 个洞的能量值修改为 b
每次丢球，输出这个球最后进入的洞的编号以及跳跃次数

---

刚开始想了一万种思路，发现都没法维护这个修改
后来看了下题解，发现是分块暴力搞……
果然分块的境界还不够啊，以后像这种维护不了的东西果断要想到分块啊……

将 N 个洞按 N‾‾√ 的大小分块，
每个洞维护一个跳出这一块后第一个进入的洞 nxt[i]
以及跳出这一块之前的跳跃次数 cnt[i]
每次修改都暴力重新维护一下一整个块，复杂度 (N‾‾√)
每次询问都顺着 nxt[i]走，复杂度也是 (N‾‾√)
总的复杂度就是 O(MN‾‾√)

#pragma comment(linker, "/STACK:102400000,102400000")#include <cstdio>#include <iostream>#include <cstdlib>#include <cstring>#include <algorithm>#include <cmath>#include <cctype>#include <map>#include <set>#include <queue>#include <bitset>#include <string>#include <complex>using namespace std;typedef pair<int,int> Pii;typedef long long LL;typedef unsigned long long ULL;typedef double DBL;typedef long double LDBL;#define MST(a,b) memset(a,b,sizeof(a))#define CLR(a) MST(a,0)#define SQR(a) ((a)*(a))#define PCUT puts("\n----------")const int maxn=1e5+10;int N,M,bsiz,A[maxn],cnt[maxn],nxt[maxn];int id(int x){return x/bsiz;}void maintain(int i){    if(i+A[i]>=N)    {        cnt[i] = 1;        nxt[i] = N;    }    else    {        int j = i+A[i];        if(id(i)==id(j))        {            if(nxt[j]==N) nxt[i] = j, cnt[i] = 1;            else nxt[i] = nxt[j], cnt[i] = cnt[j]+1;        }        else        {            cnt[i] = 1;            nxt[i] = j;        }    }}int main(){    #ifdef LOCAL    freopen("in.txt", "r", stdin);//  freopen("out.txt", "w", stdout);    #endif    while(~scanf("%d%d", &N, &M))    {        CLR(cnt); CLR(nxt);        bsiz=sqrt(N);        for(int i=0; i<N; i++) scanf("%d", &A[i]);        for(int i=N-1; i>=0; i--) maintain(i);        for(int i=0,opt,a,b; i<M; i++)        {            scanf("%d%d", &opt, &a);            a--;            if(opt)            {                int sum=0, end=a;                while(a<N)                {                    sum += cnt[a];                    end = a;                    a = nxt[a];                }                printf("%d %d\n", end+1, sum);            }            else            {                scanf("%d", &A[a]);                int l=bsiz*id(a), r=min(N,bsiz*(id(a)+1));                for(int j=r-1; j>=l; j--) maintain(j);            }        }    }    return 0;}