码迷,mamicode.com
首页 > 其他好文 > 详细

【题解】HDU-2222 Keywords Search

时间:2018-07-30 13:32:45      阅读:141      评论:0      收藏:0      [点我收藏+]

标签:memset   def   int   stl   printf   自动机   space   没有   false   

题目链接:HDU-2222  或  Vjudge

简单说明:

  ac自动机的建立,其中插入过程借助了字典树,处理回溯数组(也有人称失败数组)过程是一个广搜运用了STL的队列(queue)。ac自动机的过程还在写。

  这是学习ac自动机的第一题,如果wa的话,那就要注意,字串结束标记是如何标记的了。它不能单单 记录,还要 计数。WA了好几次……

my codes:

 

技术分享图片
#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <queue>
#define set(a) memset(a,0,sizeof(a))
#define set_b(a) memset(a,false,sizeof(a))
using namespace std;
const int M=5e5+10;
const int N=1e6+10;

int tree[M][26];/**< tree[u][i]  表示节点u的i儿子的节点值 */
int ed[M];/**< 字串结束标记,运用比较灵活,本题要计数 */
int nxt[M];/**< 回溯数组 */
char s[N];/**< 文本串 */
int tot;/**< 节点数 */


struct ac_auto
{
    void pre_set()/**< 初始化 */
    {
        tot=1;
        set(ed);
        set(tree);
        set(nxt);
    }
    void build(char str[])/**< 基本插入操作 */
    {
        int len=strlen(str);
        int u=1;
        for(int i=0; i<len; i++)
        {
            int v=str[i]-a;
            if(!tree[u][v])
                tree[u][v]=++tot;
            u=tree[u][v];
        }
//    printf("%d\n",u);
        ed[u]++;
    }
    void deal_next()/**< 处理回溯数组,用了一个虚拟的零号节点,下面两行表示它的所有儿子与一号建边 */
    {
        for(int i=0; i<26; i++)
            tree[0][i]=1;
        queue<int> q;
        while(!q.empty())
            q.pop();
        nxt[1]=0;/**< 一号节点回溯到零号 */
        q.push(1);/**< 用到了零号节点,先压入 1 。如果不设零号的话,是压入 1 的所有儿子。两种,注意区别 */
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            for(int i=0; i<=25; i++)
            {
                if(!tree[u][i])/**< u没有i这个儿子 */

                    tree[u][i]=tree[nxt[u]][i];
                                                  /** \brief u 与它的i 儿子之间建边。有了这句话那么每个节点都将
                                                            有26个儿子,就不会存在 某节点没有 i 儿子还要继续找
                                                            父亲的回溯节点的信息这一麻烦。
                                                      \return
                                                    */
                else
                {
                    nxt[tree[u][i]]=tree[nxt[u]][i];/**< u的i儿子nxt值是next[u]的i儿子的号码 */
//                printf("nxt[%d]=%d\n",tree[u][i],tree[nxt[u]][i]);
                    q.push(tree[u][i]);
                }
//            printf("%d 的 %c 儿子编号:%d\n",u,‘a‘+i,tree[u][i]);
            }
        }
    }
    int search()
    {
        int len=strlen(s);
        int ans=0;
        int u=1,v;
        for(int i=0; i<len; i++)
        {
            v=s[i]-a;
            u=tree[u][v];
            int tp=u;
            while(tp>1&&ed[tp])
            {
                ans+=ed[tp];
                ed[tp]=0;
                tp=nxt[tp];
            }
        }
        return ans;
    }
};

int main()
{
    int T,n;
    cin>>T;
    while(T--)
    {
        ac_auto().pre_set();
        cin>>n;
        while(n--)
        {
            char str[55];
            scanf("%s",str);
            ac_auto().build(str);
        }
        scanf("%s",s);
        ac_auto().deal_next();
        int ans=ac_auto().search();
        cout<<ans<<endl;
    }
    return 0;
}
View Code

 

【题解】HDU-2222 Keywords Search

标签:memset   def   int   stl   printf   自动机   space   没有   false   

原文地址:https://www.cnblogs.com/Ycourage/p/9371230.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!