BZOJ 2002, 彈飛綿羊

Problem

傳送門

Mean

輸入鏈上各彈力裝置的彈力系數，要求支持修改彈力系數以及查詢綿羊被彈幾次後彈飛。

Analysis

LCT可做，而且時間複雜度優。
分塊雖然慢一些但是編程複雜度比較低。（其實是我不想學LCT……）
兩個數組cnt和nxt分別記錄處於當前裝置時被彈幾次後彈出當前塊以及彈到後面的什麼位置。
分塊對細節處理要求比較高，題目中也特別強調了裝置編號爲0到n-1，需要注意。

Code

#include<cstdio>
#include<cmath>
const int N=200005;
int n,m,i,j,k,len,p,q=1,a[N],l[450],r[450],cnt[N],nxt[N],c[N];
int min(int a,int b){return a<b?a:b;}
int init(int x){
    if(cnt[x]) return cnt[x];
    if(x+a[x]>r[c[x]]){
        nxt[x]=x+a[x]>n?n:x+a[x];
        return cnt[x]=1;
    }
    cnt[x]=init(x+a[x])+1;
    nxt[x]=nxt[x+a[x]];
    return cnt[x];
}
void change(int x,int y){
    if(x+y>r[c[x]]) nxt[x]=x+y>n?n:x+y,cnt[x]=1;
    else nxt[x]=nxt[x+y],cnt[x]=cnt[x+y]+1;
    for(int i=l[c[x]];i<x;i++) cnt[i]=0;
    for(int i=l[c[x]];i<x;i++) if(!cnt[i]) init(i);
}
int query(int x){
    if(x>=n) return 0;
    return cnt[x]+query(nxt[x]);
}
int main(){
    scanf("%d",&n);
    for(i=0;i<n;i++) scanf("%d",&a[i]);
    len=(int)pow(n,0.5);
    while(p<n) l[q]=p,r[q++]=min((p+=len)-1,n-1);
    for(int i=1;i<q;i++) for(int j=l[i];j<=r[i];j++) c[j]=i;
    for(int i=0;i<n;i++) if(!cnt[i]) init(i);
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&i,&j);
        if(i==2){
            scanf("%d",&k);
            change(j,a[j]=k);
        }else printf("%d\n",query(j));
    }
    return 0;
}