【洛谷】弹飞绵羊-分块/LCT

传送门：洛谷：[HNOI2010]弹飞绵羊

---

题意

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

---

输入

第一行包含一个整数n，表示地上有n个装置，装置的编号从0到n-1。

接下来一行有n个正整数，依次为那n个装置的初始弹力系数。

第三行有一个正整数m，

接下来m行每行至少有两个数i、j，若i=1，你要输出从j出发被弹几次后被弹飞，若i=2则还会再输入一个正整数k，表示第j个弹力装置的系数被修改成k。

---

输出

对于每个i=1的情况，你都要输出一个需要的步数，占一行。

---

数据范围

对于20%的数据n,m<=10000，对于100%的数据n<=200000,m<=100000

---

题解

其实很好理解，分成$\sqrt{n}$个块，处理从前一个块到下一个块的个数和到下一个块的那个点，网上讲解很多。
先自己打了个非常丑的代码，居然得了30分（不得不吐槽洛谷的水数据）。
然后看了黄学长的代码，emmm，非常妙。
明天补LCT做法。
然而两天后。LCT见篇尾。

---

代码

我的30分代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<vector>
using namespace std;
const int N=2e5+10;
int n,m,t[N],siz,mx,tot,lc[N];
bool vis[N];
vector<int>ss[500];

struct Node{
    int to;
    int c;
}st[N];

inline int read()
{
    char c=getchar();int x=0,t=1;
    while(c<'0' || c>'9') {if(c=='-') t=-1;c=getchar();}
    while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
    return x*t;
}

inline void add()
{
    for(int i=1;i<=n;i++){
        int k=i/mx;
        if(!vis[i]){
            if(k%mx==0 && k!=0) ss[k-1].push_back(i);
            else ss[k].push_back(i);
            vis[i]=true;
            lc[i]=i;
            st[i].c=1;
            int now=i+t[i];
            int www=i%mx==0? i:(k+1)*mx;
            for(;now<=www && now<=n;now+=t[now]){
                vis[now]=true;lc[now]=i;st[i].c++;
            }
            if(now<=n){
                st[i].to=now;
            } 
        }
    }
}

inline int query(int k)
{
    int ans=1;
    int w=lc[k];
    while(w!=k){
        w+=t[w];ans++;
    }
    ans=st[lc[k]].c-ans;
    w=st[lc[k]].to;
    while(w!=0){
        int pw=lc[w],aa=0;
        while(pw!=w){
            pw+=t[pw];aa++;
        }
        ans+=st[lc[w]].c-aa;
        w=st[w].to;
    }
    return ans;
}

inline void change(int num,int v)
{
    int k=num/mx,o=num,ti=0;t[num]=v;
    int www= num%mx==0? num:(k+1)*mx;
    for(;o<=www && o<=n;o+=t[o]){
        ti++;
    }
    if(lc[num]==num){
        st[num].to=o>n? 0:o;
        st[num].c=ti;
    }
    k-=num%mx==0 && k!=0?1:0;
    int len=ss[k].size();
    for(int i=0;i<len;i++){
        int temp=ss[k][i],ptr=0;
        int nn=temp;
        while(nn!=st[temp].to){
            ptr++;
            nn+=t[nn];
            if(nn==num){
                st[temp].to=o>n? 0:o;
                st[temp].c=ptr+ti;
                break;
            }
        }
    }
}

int main(){
    memset(vis,false,sizeof(vis));
    n=read();mx=(int)(sqrt(n));
    for(int i=1;i<=n;i++){
        t[i]=read();
    }
    add();m=read();
    while(m--){
        int op=read();
        if(op==1){
            printf("%d\n",query(read()));
        }else{
            int num=read()+1,val=read();
            change(num,val);
        }
    }
    //system("pause");
    return 0;
}

//anyway luogu->30 st->0 

AC代码：
分块

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=2e5+10;
int n,t[N],m,mx;
int st[N],ed[N],b[N];
int l[500];

inline int read()
{
    char c=getchar();int x=0,t=1;
    while(c<'0' || c>'9') {if(c=='-') t=-1;c=getchar();}
    while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
    return x*t;
}


inline int query(int k)
{
    int ans=0;
    while(k!=0){
        ans+=st[k];
        k=ed[k];
    }
    return ans;
}

inline void add(int k,int v)
{
    t[k]=v;
    for(int i=k;i>=l[b[k]];i--){
        st[i]=1;
        if(i+t[i]>n) {
           ed[i]=0;continue;    
        }
        if(b[i]==b[i+t[i]]){
            st[i]+=st[i+t[i]];
            ed[i]=ed[i+t[i]];
        }else{
            ed[i]=i+t[i];
        }
    }
}

int main(){
    n=read();mx=(int)(sqrt(n));
    for(int i=1;i<=n;i++) t[i]=read();
    int cnt=n/mx;
    if(n%mx!=0) cnt++;
    for(int i=0;i<cnt;i++)
     l[i]=i*mx+1;
    for(int i=n;i>=1;i--){
        b[i]=(i-1)/mx;st[i]=1;
        if(i+t[i]>n) continue;
        else if(b[i]==b[i+t[i]]){
            st[i]+=st[i+t[i]];ed[i]=ed[i+t[i]];
        }else ed[i]=i+t[i];
    }
    m=read();
    while(m--){
        int op=read();
        if(op==1){
            printf("%d\n",query(read()+1));
            //query(read())-1 -> query(read()+1)
            //WA->AC
        }else{
            int p=read()+1,q=read();
            //p->题意上的问题 
            add(p,q);
        }
    }
    return 0;
}

LCT

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=2e5+10;
int n,m,to[N],t[N];
int f[N],rev[N],c[N][2],siz[N];
int now[N];
inline int read(){
    char ch=getchar();int x=0;
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x;
}

inline bool isroot(int x)
{
    return (c[f[x]][0]!=x && c[f[x]][1]!=x);
}

inline void pushup(int k)
{
    siz[k]=siz[c[k][0]]+siz[c[k][1]]+1;
}

inline void pushdown(int k)
{
    if(rev[k]){
        rev[c[k][0]]^=1;
        rev[c[k][1]]^=1;
        rev[k]=0;
        swap(c[k][0],c[k][1]);
    }
}

inline void rotate(int x)
{
    int y=f[x];int z=f[y];
    if(!isroot(y)){
        c[z][c[z][1]==y]=x;
    }
    int ptr= c[y][1]==x; 
    f[x]=z;f[y]=x;f[c[x][ptr^1]]=y;
    c[y][ptr]=c[x][ptr^1];c[x][ptr^1]=y; 
    pushup(y);pushup(x); 
}

inline void splay(int x)
{
    int tot=0,p=x;now[++tot]=x;
    for(;!isroot(p);p=f[p]) now[++tot]=f[p];
    for(int i=tot;i>0;i--) pushdown(now[i]);
    while(!isroot(x)){
        int y=f[x];int z=f[y];
        if(!isroot(y)){
            if(c[z][0]==y ^ c[y][0]==x) rotate(x);
            else rotate(y);
        }
        rotate(x);
    }
}

inline void access(int x)
{
    int t=0;
    while(x){
        splay(x);
        c[x][1]=t;
        t=x;x=f[x];
    }
}

inline void rever(int x)
{
    access(x);splay(x);rev[x]^=1;
}

inline int query(int x)
{
    rever(n+1);access(x);splay(x);
    return siz[c[x][0]];
}

inline void cut(int x,int y)
{
    rever(x);access(y);splay(y);c[y][0]=f[x]=0; 
}

inline void link(int x,int y)
{
    rever(x);f[x]=y;splay(x);
}

inline void update(int x,int val)
{
    int p=min(n+1,x+val);
    cut(x,to[x]);link(x,p);
    to[x]=p; 
}

int main(){
    n=read();
    for(int i=1;i<=n;i++) {
       t[i]=read();
       to[i]= (i+t[i])>n+1? n+1:i+t[i];
       f[i]=to[i];  
    }
    //for(int i=1;i<=n;i++) siz[i]=1;
    m=read();
    while(m--){
        int op=read();
        if(op==1){
            printf("%d\n",query(read()+1));
        }else{
            int x=read()+1,rp=read();
            update(x,rp); 
        }
    } 
    return 0;
} 

//读错题了orzz
//怪不得一直想不通为什么要read()+1
//是从编号为read()的开始
//laji题面