bzoj 3172 [Tjoi2013]单词后缀自动机

最新推荐文章于 2018-09-01 09:25:59 发布

原创最新推荐文章于 2018-09-01 09:25:59 发布 · 405 阅读

0 ·

CC 4.0 BY-SA版权

后缀自动机专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了一种使用后缀自动机解决字符串匹配问题的方法。通过构建特殊的后缀自动机结构，可以高效地处理多个字符串的匹配问题。文章提供了一个完整的实现示例，包括后缀自动机的构建过程及如何利用该结构进行查询。

把所有串放在一起，两个串中间加非法字符，建一个后缀自动机。
然后按拓扑序遍历自动机，求一个点子树中的结尾数量。
然后对于每个串，在后缀自动机中跑trs指针找到对应终点。
那么这个串的答案就是终点子树中的结尾数量。

#include <bits/stdc++.h>
using namespace std;
#define N 1100000
int n,len,l[210],beg[210];
char s[N];
struct SAM
{
    int trs[N<<1][28],fa[N<<1],len[N<<1],num[N<<1];
    int du[N<<1],cnt,last;
    queue<int>q;
    void init(){cnt=last=1;}
    void insert(int x)
    {
        int p=last,np=++cnt,q,nq;
        last=np;len[np]=len[p]+1;num[np]++;
        for(;p&&!trs[p][x];p=fa[p])trs[p][x]=np;
        if(!p)fa[np]=1;
        else
        {
            q=trs[p][x];
            if(len[q]==len[p]+1)fa[np]=q;
            else
            {
                fa[nq=++cnt]=fa[q];
                len[nq]=len[p]+1;
                memcpy(trs[nq],trs[q],sizeof(trs[nq]));
                fa[q]=fa[np]=nq;
                for(;p&&trs[p][x]==q;p=fa[p])trs[p][x]=nq;
            }
        }
    }
    void cal()
    {
        for(int i=1;i<=cnt;i++)du[fa[i]]++;
        for(int i=1;i<=cnt;i++)
            if(!du[i])q.push(i);
        while(!q.empty())
        {
            int t=q.front();q.pop();
            if((--du[fa[t]])==0)q.push(fa[t]);
            num[fa[t]]+=num[t];
        }
    }
}sam;
int main()
{
    //freopen("tt.in","r",stdin);
    sam.init();
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+len+1);
        beg[i]=len+1;
        len+=(l[i]=strlen(s+len+1));
        s[++len]='z'+1;
    }
    for(int i=1;i<=len;i++)
        sam.insert(s[i]-'a');
    sam.cal();
    for(int i=1;i<=n;i++)
    {
        int now=1;
        for(int j=beg[i];j<=beg[i]+l[i]-1;j++)
            now=sam.trs[now][s[j]-'a'];
        printf("%d\n",sam.num[now]);
    }
    return 0;
}