【bzoj 2002】弹飞绵羊

传送门~

解题思路

可以LCT，分块稳稳水过。
分$\sqrt{n}$块，f[i]表示跳几次跳出当前所在块，g[i]表示跳出块以后落在哪一个节点。块内从后向前递推求出f[i]和g[i]，修改和查询复杂度均为$O(\sqrt(n))$。
分块代码：

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
using namespace std;
int f[200005],g[200005];
int block[200005],ki[200005];
int lo,n,m;
void idt(){
    int r=sqrt(n),ld=1;
    block[1]=1;lo=1;
    for(int i=1;i<=n;i++){
        if(i-ld>r) {ld=i;lo++;block[i]=lo;}
        else block[i]=block[ld];
    }

    for(int i=n;i>=1;i--){
        if(i+ki[i]<=n && block[i]==block[i+ki[i]]){
            f[i]=f[i+ki[i]]+1;g[i]=g[i+ki[i]];
        }
        else {
            f[i]=1;g[i]=i+ki[i];    
        }
    }

} 
void res(int a,int b){
    ki[a]=b;
    for(int i=a;i>=1 && block[a]==block[i];i--){
        if(i+ki[i]<=n && block[i]==block[i+ki[i]]){
            f[i]=f[i+ki[i]]+1;g[i]=g[i+ki[i]];
        }
        else {
            f[i]=1;g[i]=i+ki[i];    
        }
    }
}
void solve(int a){
    int j=0;
    while(a<=n){
        j+=f[a];
        a=g[a];
    }
    printf("%d\n",j);
}
int main(){
    int opt,a,b;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&ki[i]);
    idt();scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d",&a);
            solve(a+1);
        }
        else {
            scanf("%d%d",&a,&b);
            res(a+1,b);
        }
    }
    return 0;
}

LCT代码：

#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
#include<string>
#include<cstdio>
#include<cstdlib>
using namespace std;
struct node{
    int key,siz;
    node* ch[2],*fa;
    node();
    void maintain();
    int son(){
        if(fa->ch[0]==this) return 0; 
        if(fa->ch[1]==this) return 1;
        return -1;
    }
}*null=new node(),*s[200005];
node:: node(){
    siz=null ? 1 : 0;
    ch[0]=ch[1]=fa=null;
}
void node:: maintain() {siz=ch[0]->siz+ch[1]->siz+1;}
void Rotate(node* p,bool f){
    node* t=p->ch[f^1];
    p->ch[f^1]=t->ch[f];
    if(t->ch[f]!=null) t->ch[f]->fa=p;
    t->ch[f]=p;
    p->maintain();t->maintain();
    if(~p->son()) p->fa->ch[p->son()]=t;
    t->fa=p->fa;p->fa=t;
}
void splay(node* p){
    while(~p->son()){
        int dir=p->son();
        if(dir==p->fa->son()) Rotate(p->fa->fa,dir^1);
        Rotate(p->fa,dir^1);
    }
}
void Access(node *p){
    node *t=null;
    while(p!=null){
        splay(p);
        p->ch[1]=t;p->maintain();
        t=p;p=p->fa;
    }
}
int solve(node* p){
    Access(p);splay(p);
    return p->ch[0]->siz;
}
int main(){
    int n,m;scanf("%d",&n);
    null->ch[0]=null->ch[1]=null->fa=null;
    for(int i=1;i<=n+1;i++) {s[i]=new node();s[i]->key=i;}
    for(int i=1;i<=n;i++){
        int x;scanf("%d",&x);
        x=min(n+1,i+x);
        s[i]->fa=s[x];
    }
    scanf("%d",&m); 
    for(int i=1;i<=m;i++){
        int opt,x,y;
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d",&x);x++;
            printf("%d\n",solve(s[x]));
        }
        else {
            scanf("%d%d",&x,&y);x++;
            Access(s[x]);splay(s[x]);
            s[x]->ch[0]->fa=s[x]->fa;
            s[x]->ch[0]=s[x]->ch[1]=null;
            s[x]->maintain();
            y=min(n+1,x+y);s[x]->fa=s[y];
        }
    }
    return 0;
}