HDU 2222 AC自动机

简介:

题意:给出多组模式串,再给出一个母串,问多少模式串是母串的字串。

裸AC自动机题,来试模板的。

#include <iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<queue>
using namespace std;
char key[55],des[1000005];
const int MAX_NODE = 1000005;
const int SIGMA_SIZE = 26;
int size;
struct node
{
    node* fail;
    node* next[SIGMA_SIZE];
    int count;
    void init()
    {
        fail=NULL;
        count=0;
        memset(next, 0, sizeof(next));
    }
};
class AC_Automation
{
public:
    void init();
    void insert(char* str);
    void getFail();
    int find(char* str);
private:
    node* new_node();
    node Heap[MAX_NODE];
    node* root;
    int size;
};
node* AC_Automation::new_node()
{
    Heap[size].init();
    return &Heap[size++];
}
void AC_Automation::init()
{
    size=0;
    root=new_node();
}
void AC_Automation::insert(char* str)
{
    node* p=root;
    for( ; *str; ++str)
    {
        int ch=*str-'a';
        if(p->next[ch]==NULL)
            p->next[ch] = new_node();
        p=p->next[ch];
    }
    ++p->count;
}
void AC_Automation::getFail()
{
    queue<node*>Q;
    Q.push(root);
    while(!Q.empty())
    {
        node* tmp=Q.front();
        Q.pop();
        node* p;
        for(int i=0; i<SIGMA_SIZE; ++i)
        {
            node*& now=tmp->next[i];
            if(now != NULL)
            {
                Q.push(now);
                if(tmp==root)
                {
                    now->fail=root;
                    continue;
                }
                p = tmp->fail;
                while(p != NULL)
                {
                    if(p->next[i] != NULL)
                    {
                        now->fail = p->next[i];
                        break;
                    }
                    p=p->fail;
                }
                if(p==NULL) now->fail=root;
            }
            else
            {
                if(tmp==root) now=root;
                else now=tmp->fail->next[i];
            }
        }
    }
}
int AC_Automation::find(char* str)
{
    node* p=root;
    int cnt=0;
    for( ; *str; ++str)
    {
        char ch=*str-'a';
        p = p->next[ch];
        if(p==NULL) p=root;
        node* tmp=p;
        while(tmp!=root && tmp->count!=-1)
        {
            cnt += tmp->count;
            tmp->count=-1;
            tmp=tmp->fail;
        }
    }
    return cnt;
}
AC_Automation ac;
int main()
{
    int n,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        ac.init();
        for(int i=0; i<n; i++)
            scanf("%s",key),ac.insert(key);
        ac.getFail();
        scanf("%s",des);
        printf("%d\n",ac.find(des));
    }
    return 0;
}


目录
相关文章
|
22天前
|
算法
Leetcode第十四题(最长公共前缀)
这篇文章介绍了一种算法,用于在给定的字符串数组中找到最长公共前缀,通过逐字符比较每个字符串的对应位置,一旦发现不匹配立即返回当前已匹配的子串作为公共前缀。
23 0
|
22天前
|
机器学习/深度学习 算法 C++
Leetcode第52题(N皇后II)
这篇文章介绍了解决LeetCode第52题(N皇后II)的C++代码实现,使用深度优先搜索(DFS)算法来找出所有可能的解决方案,并计算出不同的解决方案数量。
12 1
|
18天前
acwing 1113 红与黑
acwing 1113 红与黑
7 0
|
22天前
Leetcode第十八题(四数之和)
这篇博客介绍了LeetCode第18题“四数之和”的解法,通过排序和双指针技术来找出数组中所有和为特定值的四个不同元素的组合。
10 0
|
21天前
lanqiao OJ 641 迷宫
lanqiao OJ 641 迷宫
26 0
|
21天前
lanqiao OJ 234 大胖子走迷宫
lanqiao OJ 234 大胖子走迷宫
9 0
|
22天前
|
存储 算法 C++
Leetcode第三十六题(有效的数独)
这篇文章介绍了如何使用C++编写一个算法来验证一个9x9数独是否有效,遵循数独的规则,即数字1-9在每一行、每一列和每个3x3宫内只能出现一次。
34 0
|
6月前
|
人工智能
蛇形填数(蓝桥杯)
蛇形填数(蓝桥杯)
|
6月前
|
机器学习/深度学习
leetcode-52:N皇后 II
leetcode-52:N皇后 II
26 0
|
数据采集 算法 数据挖掘
【每周一坑】螺旋矩阵
今天这题,看起来挺简单,实际写出来并不容易。在以前公司我曾把它做过招聘的笔试题,结果惨不忍睹,不得不拿掉。
【每周一坑】螺旋矩阵