HDU 2222（AC自动机模板题)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=2222

题目大意：多个模式串。问匹配串中含有多少个模式串。注意模式串有重复，所以要累计重复结果。

解题思路：

AC自动机模板题。

一开始使用LRJ的坑爹静态模板，不支持重复的模式串。

在做AC自动机+DP的时候，扒了zcwwzdjn大神的动态优化（失配指向root）写法，以及借鉴了网上的AC自动机模板，

搞出了这么一个支持重复串的模板。

注意在计算last后缀链接的时候，都会修改last->cnt=0，是为了防止同一个模式串在匹配时候走了多遍。比如模：AA，匹：AAA,

修改last->cnt的结果是1，而不修改就是2。修改法在碰到HDU 2896这种需要计算多个匹配串的时候就萎了。

必须在修改之后，及时恢复。由于本身就使用了queue，所以修改时用queue记录一下哪些last被修改了，之后恢复一下就OK！

同时，修改法在HDU 3065中，统计一个模式串出现多少次也萎了。因为模式串必须要走多遍。

 

#include "cstdio"

#include "cstring"

#include "iostream"

#include "queue"

#include "string"

using namespace std;

struct Trie

{

    Trie *next[26],*fail;

    int cnt;

}*root;

struct status

{

    Trie *last;

    int cnt;

    status(Trie *last,int cnt):last(last),cnt(cnt) {}

};

Trie *newnode()

{

    Trie *ret=new Trie;

    memset(ret->next,0,sizeof(ret->next));

    ret->fail=0;

    ret->cnt=0;

    return ret;

}

void init() {root=newnode();}

void Insert(string str)

{

    Trie *pos=root;

    for(int i=0;i<str.size();i++)

    {

        int c=str[i]-'a';

        if(!pos->next[c]) pos->next[c]=newnode();

        pos=pos->next[c];

    }

    pos->cnt++;

}

void getfail()

{

   queue<Trie *> Q;

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

   {

       if(root->next[c])

       {

           root->next[c]->fail=root;

           Q.push(root->next[c]);

       }

       else root->next[c]=root;

   }

   while(!Q.empty())

   {

       Trie *x=Q.front();Q.pop();

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

       {

           if(x->next[c])

           {

               x->next[c]->fail=x->fail->next[c];

               Q.push(x->next[c]);

           }

           else x->next[c]=x->fail->next[c];

       }

   }

}

int find(string str)

{

    Trie *pos=root,*last;

    queue<status> Q;

    int ans=0;

    for(int i=0;i<str.size();i++)

    {

        int c=str[i]-'a';last;

        if(pos->next[c])

        {

            pos=pos->next[c];

            last=pos;

            while(last->cnt)

            {

                Q.push(status(last,last->cnt));

                ans+=last->cnt;

                last->cnt=0; //修改last->cnt

                last=last->fail;

            }

        }

    }

    while(!Q.empty()) //恢复last->cnt

    {

        status x=Q.front();Q.pop();

        x.last->cnt=x.cnt;

    }

    return ans;

}

int main()

{

    //freopen("in.txt","r",stdin);

    ios::sync_with_stdio(false);

    int T,n;

    string tt;

    cin>>T;

    while(T--)

    {

        cin>>n;

        init();

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

        {

            cin>>tt;

            Insert(tt);

        }

        getfail();

        cin>>tt;

        int ans=find(tt);

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

    }

}

 

11927003

2014-10-21 01:22:37

Accepted

2222

390MS

29480K

1942 B

C++

Physcal