luogu4036 [JSOI2008]火星人(splay+二分hash)

題面在這裏

題意

給出一個字符串,要求茲瓷動態插入/修改字符,詢問x,y開頭的後綴的lcp。保證字符串任意時刻長度 105

做法

splay維護區間hash值。

查詢時二分+hash即可。

代碼

=> 不得不吐槽bzoj的評測機速度 -.- 果斷地去luogu上提交。

=> 注意splay的一些細節。以及二分的邊界r注意取min,大概題中不保證 xy

#include<bits/stdc++.h>
#define rep(i,x,y) for (int i=(x); i<=(y); i++)
#define ll long long
#define ld long double
#define inf 1000000000
using namespace std;
#define N 100005
#define mod 20030731
#define seed 27
int n,m,tot,rt,pw[N],fa[N],siz[N],ch[N][2],data[N]; ll h[N]; char s[N],opt[5];
void up(int x){
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
    h[x]=(h[ch[x][1]]+(ll)data[x]*pw[siz[ch[x][1]]]%mod+(ll)h[ch[x][0]]*pw[siz[ch[x][1]]+1]%mod)%mod;
}
void rot(int x){
    int y=fa[x],z=fa[y],f=ch[y][1]==x;
    ch[y][f]=ch[x][f^1]; if (ch[x][f^1]) fa[ch[x][f^1]]=y;
    fa[x]=z; if (z) ch[z][ch[z][1]==y]=x;
    fa[y]=x; ch[x][f^1]=y; up(y),up(x);
}
void splay(int x,int tp){
    while (fa[x]!=tp){
        int y=fa[x],z=fa[y];
        if (z!=tp) rot((ch[y][0]==x)==(ch[z][0]==y)?y:x);
        rot(x);
    }
    if (!tp) rt=x;
}
int build(int pr,int l,int r){
    if (l>r) return 0;
    int k=++tot,mid=l+r>>1;
    fa[k]=pr; siz[k]=1;
    data[k]=h[k]=s[mid]-'a';
    if (l!=r){
        ch[k][0]=build(k,l,mid-1),ch[k][1]=build(k,mid+1,r);
        up(k);
    }
    return k;
}
int getkth(int k){
    if (!k) return 0;
    int x=rt;
    while (1){
        if (k<=siz[ch[x][0]]) x=ch[x][0];
        else if (k>siz[ch[x][0]]+1) k-=siz[ch[x][0]]+1,x=ch[x][1];
        else return x;
    }
}
int gethash(int x,int y){
    x++,y++; x=getkth(x-1),y=getkth(y+1);
    splay(x,0); splay(y,x);
    return h[ch[y][0]];
}
void ins(int x,char chr){
    x++; int y=getkth(x),z=getkth(x+1);
    splay(y,0); splay(z,y);
    int &t=ch[z][0]; t=++tot;
    fa[t]=z; siz[t]=1; data[t]=h[t]=chr;
    up(z); up(y);
}
void mdf(int x,char chr){
    x++; x=getkth(x); splay(x,0);
    data[x]=chr; up(x);
}
int main(){
    scanf("%s",s+2); n=strlen(s+2);//維護區間hash值
    pw[0]=1; rep (i,1,N-1) pw[i]=(ll)seed*pw[i-1]%mod;
    rt=build(0,1,n+2); tot=n+2;
    scanf("%d",&m);
    while (m--){
        scanf("%s",opt); int x,y;
        if (opt[0]=='Q'){
            scanf("%d%d",&x,&y);
            int l=1,r=min(tot-y-1,tot-x-1),mid;//注意r這裏一定要取min!
            while (l<=r){
                mid=l+r>>1;
                if (gethash(x,x+mid-1)==gethash(y,y+mid-1)) l=mid+1; else r=mid-1;
            }
            printf("%d\n",l-1);
        } else if (opt[0]=='I'){
            scanf("%d%s",&x,s); ins(x,s[0]-'a');
        } else{
            scanf("%d%s",&x,s); mdf(x,s[0]-'a');
        }
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章