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;
}