hdu 3065 病毒侵袭持续中 DFA 求字符串最多匹配多少字符串 每个子串最多匹配多少个

#include<iostream>

#include<cstdio>

#include<cstring>

#include<string>

#include<queue>

#include<cctype>

#include<algorithm>

using namespace std;

//DFA 求字符串最多匹配多少字符串 每个子串最多匹配多少个

const int maxn=2100000;

struct Node

{

    int num;//序号

    Node *next[27];//后继指针

    Node* fail;//失败指针

    /*指向树中出现过的S的最长的后缀,换句话说就是在前缀集中出现的最长的S的后缀.*/

};

//字典树

Node temp[maxn];

int tp;

//.........

int n;//字符串个数

Node *root;//头指针

char mat[1100][55];//模式串 各不相同

int cont[1100];//模式串在主串中出现次数

void reset(Node* rot)

{

    rot->num=0;

    for(int i=0;i<27;i++) rot->next[i]=NULL;

    rot->fail=root;//important

    if(rot==root) rot->fail=NULL;//important

}

//初始化

void init()

{

memset(cont,0,sizeof(cont));

    tp=0;

    root=&temp[tp++];

    reset(root);

}

//插入字典树

void insert(char *word,int index)

{

    Node *p=root;

    for(int i=0;word[i];i++)

    {

        int x;

        if(isupper(word[i]))  x=word[i]-'A';

        else x=26;

        if(p->next[x]==NULL)

        {

            p->next[x]=&temp[tp++];

            reset(p->next[x]);

        }

        p=p->next[x];

    }

    p->num=index;

}

//构造失败指针

Node *que[maxn*4];

int front,rear;

void DFA()

{

    Node* p=root;

    front=rear=0;

    que[rear++]=p;

    while(front<rear)

    {

        Node* t=que[front++];

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

        {

            Node* cnt=t->next[i];

            if(cnt!=NULL)

            {

                Node* fath=t->fail;

                while(fath!=NULL&&fath->next[i]==NULL)

                {

                    fath=fath->fail;

                }

                if(fath!=NULL)

                {

                    cnt->fail=fath->next[i];

                }

                else

                {

                    cnt->fail=p;//important

                }

                que[rear++]=cnt;

            }

        }

    }

}

//查看str最多匹配多少字符串 每个子串最多匹配多少个

void find(char* str)

{

    Node* p=root;

    Node* fath;

    Node* cnt;

    for(int i=0;str[i];i++)

    {

        int x;

        if(isupper(str[i])) x=str[i]-'A';

        else x=26;

        cnt=p->next[x];

        if(cnt!=NULL)

        {

            for(fath=cnt;fath!=NULL;fath=fath->fail)

            {

                if(fath->num)

                {

                    cont[fath->num]++;

                }

            }

            p=cnt;

        }

        else

        {

            fath=p->fail;

            while(fath!=NULL&&fath->next[x]==NULL)

            {

                fath=fath->fail;

            }

            if(fath!=NULL)

            {

                cnt=fath->next[x];

                for(fath=cnt;fath!=NULL;fath=fath->fail)

                {

                    if(fath->num)

                    {

                        cont[fath->num]++;

                    }

                }

                p=cnt;

            }

            else

            {

                p=root;

            }

        }

    }

}

 

char str[maxn];

int main()

{

    while(scanf("%d ",&n)==1)

    {

        init();//初始化

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

        {

            gets(mat[i]);

            memcpy(str,mat[i],sizeof(mat[i]));

            insert(str,i);//在字典树上插入字符串

        }

        DFA();//构造失败指针

        gets(str);//主串

        find(str);

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

        {

        if(cont[i]) printf("%s: %d/n",mat[i],cont[i]);

        }

    }

    return 0;

}