trie图模板

版权声明:版权声明:本文为博主原创文章,未经博主允许不得转载,欢迎添加友链。 https://blog.csdn.net/zzk_233/article/details/82870672

AC自动机链接:https://blog.csdn.net/zzk_233/article/details/82855116

基于AC自动机的模板基础上,因为AC自动机一旦失配并且在没有fail指针的时候只能回到0,但是有些时候可以在另一个串上可以

找到下一位匹配的串,所以可以给没有子节点的串指一个to指向父节点失配指针的子节点,这样别人查到它的时候,就可以有失配

指针了。

            if(t[root][i])
            {
                fail[t[root][i]]=t[fail[root]][i];
                M.push(t[root][i]);
            }else
            {
                t[root][i]=t[fail[root]][i];
            }

跟AC自动机模板(上方链接)一样的题的题解,trie图版本

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
char s[155][1001];
char s1[1000006];
struct node
{
    int ch[27];
    int fail;
    int val;
    int num;
}t[155*1001];
int sz,root;
void insert(char *ss,int ww)
{
    int now=0,l=strlen(ss+1);
    for(int i = 1;i <= l;i++)
    {
        int w=ss[i]-'a';
        if(!t[now].ch[w])t[now].ch[w]=++sz;
        now=t[now].ch[w];
    }
    t[now].val++;t[now].num=ww;
}
int q[155*1001],head,tail;
int used[155];
struct tt
{
    int tot;
    int pos;
}ans[155];
queue<int>M;
void get_fail()
{
    int root=0;
    for(int i=0;i<26;i++)
    {
        if(t[root].ch[i])M.push(t[root].ch[i]);
    }
    while(!M.empty())
    {
        root=M.front();M.pop();
        for(int i=0;i<26;i++)
        {
            if(t[root].ch[i])
            {
                t[t[root].ch[i]].fail=t[t[root].fail].ch[i];
                M.push(t[root].ch[i]);
            }else
            {
                t[root].ch[i]=t[t[root].fail].ch[i];
            }
        }
    }
}
void query(char *ss)
{
    int now=root,l=strlen(ss);
    for(int i = 0;i < l;i++)
    {
        int w=ss[i]-'a';
        while(now!=root&&!t[now].ch[w])now=t[now].fail;
        if(t[now].ch[w])
        {
            now=t[now].ch[w];
            for(int j=now;j!=root/*&&!used[j]*/;j=t[j].fail)
            {
                ans[t[j].num].tot+=t[j].val;//used[j]=1;
            }
        }
    }
}
int cmp(tt x,tt y)
{
    if(x.tot==y.tot)return x.pos<y.pos;
    return x.tot>y.tot;
}
int main()
{
    //freopen("tt.out","w",stdout);
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)break;
        memset(t,0,sizeof(t));sz=0;memset(ans,0,sizeof(ans));memset(q,0,sizeof(q)); 
        for(int i = 1;i <= n;i++)
        {
            scanf("%s",s[i]+1);
            insert(s[i],i);
            ans[i].pos=i;
        }
        get_fail();
        scanf("%s",s1);
        query(s1);
        sort(ans+1,ans+1+n,cmp);
        printf("%d\n",ans[1].tot);
        if(ans[1].tot==0)continue;
        for(int i = 1;i <= n;i++)
        {
            if(ans[i].tot!=ans[i+1].tot)
            {
                printf("%s\n",s[ans[i].pos]+1);
                break;
            }
            printf("%s\n",s[ans[i].pos]+1);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/zzk_233/article/details/82870672