bzoj 2780: [Spoj]8093 Sevenk Love Oimaster【广义SAM】

AC自动机比较简单,把询问串做成AC自动机然后模板串边跑变更新即可
SAM是把模板串做成广义SAM,然后每个节点存有几个模板串经过,具体方法是每次更新暴力向上跳直到有时间戳我不会证为什么时间复杂度是对的,沿途更新个数,查询的时候直接匹配到最后的点然后输出个数即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=400005;
int n,m,fa[N],ch[N][27],dis[N],cur=1,con=1,la,si[N],v[N],ti;
char s[N];
void ins(int c,int id)
{
    la=cur,dis[cur=++con]=id;
    int p=la;
    for(;p&&!ch[p][c];p=fa[p])
        ch[p][c]=cur;
    if(!p)
        fa[cur]=1;
    else
    {
        int q=ch[p][c];
        if(dis[q]==dis[p]+1)
            fa[cur]=q;
        else
        {
            int nq=++con;
            dis[nq]=dis[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q],v[nq]=v[q],si[nq]=si[q];
            fa[q]=fa[cur]=nq;
            for(;ch[p][c]==q;p=fa[p])
                ch[p][c]=nq;
        }
    }
    p=cur;
    for(;p&&v[p]!=ti;p=fa[p])
        v[p]=ti,si[p]++;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int j=1;j<=n;j++)
    {
        scanf("%s",s+1);
        cur=1,ti++;
        for(int i=1,len=strlen(s+1);i<=len;i++)
            ins(s[i]-'a',i);
    }
    while(m--)
    {
        scanf("%s",s+1);
        int len=strlen(s+1),p=1;
        for(int i=1;i<=len&&p;i++)
            p=ch[p][s[i]-'a'];
        printf("%d\n",si[p]);
    }
    return 0;
}
posted @ 2018-11-23 19:51  lokiii  阅读(145)  评论(0编辑  收藏  举报