UVa1449 - Dominating Patterns(AC自动机)

题目大意

给定n个由小写字母组成的字符串和一个文本串T，你的任务是找出那些字符串在文本中出现的次数最多

题解

一个文本串，多个模式串，这刚好是AC自动机处理的问题

代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <map>

#include <string>

#include <queue>

using namespace std;

const int maxnode=11111;

const int sigma_size=26;

const int maxs=160;

map<string,int> ms;

struct AhoCorasickAutomata 

{

    int ch[maxnode][sigma_size],last[maxnode];

    int fail[maxnode],end[maxnode],cnt[160];

    int sz;

    void init()

    {

        memset(ch[0],0,sizeof(ch[0]));

        memset(cnt,0,sizeof(cnt));

        sz=1,ms.clear();

    }

    int idx(char c){return c-'a';}

    void insert(char *s,int v)

    {

        int u=0,n=strlen(s);

        for(int i=0;i<n;i++)

        {

            int c=idx(s[i]);

            if(!ch[u][c])

            {

                memset(ch[sz],0,sizeof(ch[sz]));

                end[sz]=0;

                ch[u][c]=sz++;

            }

            u=ch[u][c];

        }

        end[u]=v,ms[string(s)]=v;

    }

    void getfail()

    {

        queue<int>q;

        fail[0]=0;

        for(int c=0;c<sigma_size;c++)

        {

            int u=ch[0][c];

            if(u){ fail[u]=0;q.push(u);last[u]=0;}

        }

        while(!q.empty())

        {

            int r=q.front();q.pop();

            for(int c=0;c<sigma_size;c++)

            {

                int u=ch[r][c];

                if(!u){ch[r][c]=ch[fail[r]][c];continue;}

                q.push(u);

                fail[u]=ch[fail[r]][c];

                last[u] = end[fail[u]]?fail[u]:last[fail[u]];

            }

        }

    }

    void count(int j)

    {

        if(j)

        {

            cnt[end[j]]++;

            count(last[j]);

        }

    }

    void find(char *T)

    {

        int n=strlen(T);

        int j=0;

        for(int i=0;i<n;i++)

        {

            int c=idx(T[i]);

            j=ch[j][c];

            if(end[j]) count(j);

            else if(last[j]) count(last[j]);

        }

    }

};

AhoCorasickAutomata ac;

char text[1000005],P[160][80];

int main()

{

    int n;

    while(scanf("%d",&n)!=EOF&&n)

    {

        ac.init();

        for(int i=1;i<=n;i++)

        {

            scanf("%s",P[i]);

            ac.insert(P[i],i);

        }

        ac.getfail();

        scanf("%s",text);

        ac.find(text);

        int best=-1;

        for(int i=1;i<=n;i++)

            if(ac.cnt[i]>best) best=ac.cnt[i];

        printf("%d\n",best);

        for(int i=1;i<=n;i++)

            if(ac.cnt[ms[string(P[i])]]==best) printf("%s\n",P[i]);

    }

    return 0;

}