bzoj 3998 弦论 - SAM

Mys_C_K

于 2018-08-13 15:11:59 发布

阅读量189

点赞数

分类专栏：后缀自动机 BZOJ

本文链接：https://blog.youkuaiyun.com/Mys_C_K/article/details/81630375

版权

BZOJ 同时被 2 个专栏收录

138 篇文章

订阅专栏

后缀自动机

8 篇文章

订阅专栏

本文深入探讨了字符串匹配领域的Suffix Array Machine (SAM)算法实现细节。通过分析核心数据结构和操作流程，展示如何高效处理字符串匹配问题，并附带完整代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

SAM入门题

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<assert.h>
#define pb push_back
#define N 1000010
#define SIG 26
#define lint long long
#define tms(x) ((kd==0?1:sz[x])*(x!=1))
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
int rt,las,node_cnt,kd,k,sz[N],val[N],ch[N][SIG],t[N],ans[N];
lint f[N];vector<int> g[N];int fa[N],d[N];queue<int> q;
inline int new_node(int v,int x=0) { return val[x=++node_cnt]=v,x; }
inline int extend(int w)
{
    int p=las,np=new_node(val[p]+1);
    while(p&&!ch[p][w]) ch[p][w]=np,p=fa[p];
    if(!p) fa[np]=rt;
    else{
        int q=ch[p][w],v=val[p]+1;
        if(val[q]==v) fa[np]=q;
        else{
            int nq=new_node(v);
            fa[nq]=fa[q],fa[q]=fa[np]=nq;
            memcpy(ch[nq],ch[q],sizeof ch[q]);
            while(p&&ch[p][w]==q) ch[p][w]=nq,p=fa[p];
        }
    }
    return las=np,sz[np]=1;
}
int gsz(int x)
{
    for(int i=0;i<(int)g[x].size();i++)
        sz[x]+=gsz(g[x][i]);return sz[x];
}
char s[N];
int main()
{
    scanf("%s%d%d",s+1,&kd,&k),rt=las=node_cnt=1;
    int n=(int)strlen(s+1),c=0;
    for(int i=1;i<=n;i++) extend(s[i]-'a');
    for(int i=2;i<=node_cnt;i++) g[fa[i]].pb(i);
    gsz(rt);
    for(int i=1;i<=node_cnt;i++)
        for(int j=0;j<SIG;j++) if(ch[i][j]) d[ch[i][j]]++;
    for(int i=1;i<=node_cnt;i++) if(!d[i]) q.push(i);
    while(!q.empty())
    {
        int x=q.front();q.pop(),t[++c]=x;
        for(int i=0;i<SIG;i++)
            if(!(--d[ch[x][i]])) q.push(ch[x][i]);
    }
    for(int i=node_cnt,x;i>=1;i--)
    {
        f[x=t[i]]=tms(t[i]);
        for(int j=0;j<SIG;j++) f[x]+=f[ch[x][j]];
    }
    if(k>f[1]) return !printf("-1\n");
    int ansl=0,x=1;
    while(k>tms(x))
    {
        k-=tms(x);
        for(int i=0;i<SIG;i++)
            if(k>f[ch[x][i]]) k-=f[ch[x][i]];
            else { x=ch[x][i],ans[++ansl]=i;break; }
    }
    for(int i=1;i<=ansl;i++) printf("%c",ans[i]+'a');
    return !printf("\n");
}