HDU-4641 K-string (後綴自動機+線段樹合併)

HDU-4641 K-string (後綴自動機+線段樹合併)

因爲\(k\)是固定的，因此會比較方便

對於整個串構建後綴自動機

利用線段樹合併離線求出每一個點的\(endpos\)狀況，找到第\(k\)個元素，那個元素之後的的位置都出現了\(k\)次，將它們的答案加上\(len-len[link]\)，最後累前綴和便可

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

#define pb push_back
template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); }

char IO;
template<class T=int> T rd(){
    T s=0;
    int f=0;
    while(!isdigit(IO=getchar())) if(IO=='-') f=1;
    do s=(s<<1)+(s<<3)+(IO^'0');
    while(isdigit(IO=getchar()));
    return f?-s:s;
}

const int N=5e5+10,K=30;

int n,m,k;
char s[N];
int trans[N][26],link[N],len[N],stcnt,lst,End[N];

struct Edge{
    int to,nxt;
}e[N];
int head[N],ecnt;
void AddEdge(int u,int v){ 
    e[++ecnt]=(Edge){v,head[u]};
    head[u]=ecnt;
}


namespace I_Hate_It{
    int ls[N*K],rs[N*K],s[N*K],cnt;
    void Init(){ rep(i,0,cnt) ls[i]=rs[i]=s[i]=0; cnt=0; }
    void Upd(int p,int l,int r,int x) {
        s[p]++;
        if(l==r) return;
        int mid=(l+r)>>1;
        x<=mid?Upd(ls[p]?ls[p]:ls[p]=++cnt,l,mid,x):Upd(rs[p]?rs[p]:rs[p]=++cnt,mid+1,r,x);
    }
    int merge(int x,int y) {
        if(!x||!y) return x|y;
        ls[x]=merge(ls[x],ls[y]),rs[x]=merge(rs[x],rs[y]);
        s[x]=s[ls[x]]+s[rs[x]];
        return x;
    }
    int Kth(int p,int l,int r,int k){ 
        if(s[p]<k) return n+1;
        if(l==r) return l;
        int mid=(l+r)>>1;
        if(s[ls[p]]>=k) return Kth(ls[p],l,mid,k);
        else return Kth(rs[p],mid+1,r,k-s[ls[p]]);
    }
}
using I_Hate_It::Upd;
using I_Hate_It::merge;
using I_Hate_It::Kth;
using I_Hate_It::cnt;
int rt[N];

void Init(){
    link[0]=-1;
    I_Hate_It::Init();
    rep(i,0,stcnt) {
        End[i]=-1,rt[i]=head[i]=0;
        rep(j,0,25) trans[i][j]=0;
    }
    lst=stcnt=ecnt=0;
}

void Extend(int c) {
    int cur=++stcnt,p=lst;
    End[cur]=len[cur]=len[p]+1;
    while(~p && !trans[p][c]) trans[p][c]=cur,p=link[p];
    if(p==-1) link[cur]=0;
    else {
        int q=trans[p][c];
        if(len[q]==len[p]+1) link[cur]=q;
        else {
            int clone=++stcnt;
            memcpy(trans[clone],trans[q],104);
            link[clone]=link[q],len[clone]=len[p]+1;
            while(~p && trans[p][c]==q) trans[p][c]=clone,p=link[p];
            link[cur]=link[q]=clone;
        }
    }
    lst=cur;
}

int Querys[N],Qc;
ll Ans[N];

void dfs(int u){
    if(~End[u]) Upd(rt[u]=++cnt,1,n,End[u]);
    for(reg int i=head[u];i;i=e[i].nxt) {
        int v=e[i].to;
        dfs(v);
        rt[u]=merge(rt[u],rt[v]);// 線段樹合併
    }
    if(u) Ans[Kth(rt[u],1,n,k)]=len[u]-len[link[u]];  //找到第k大更新
}

int main(){
    rep(i,0,N-1) End[i]=-1;
    while(~scanf("%d%d%d",&n,&m,&k)) {
        scanf("%s",s+1);
        Init();
        rep(i,1,n) Extend(s[i]-'a');
        Qc=0;
        rep(i,1,m) {
            int opt=rd();
            if(opt==2) Querys[++Qc]=n;
            else {
                s[++n]=getchar();
                Extend(s[n]-'a');
            }
        }
        rep(i,1,stcnt) AddEdge(link[i],i);
        rep(i,0,n) Ans[i]=0;
        dfs(0);
        rep(i,1,n) Ans[i]+=Ans[i-1];//前綴和，離線預處理
        rep(i,1,Qc) printf("%lld\n",Ans[Querys[i]]);
    }
}