ac自动机(模板+一点点小优化)

题意:给定n个模式串和1个文本串,求有多少个模式串在文本串里出现过。
题解:ac自动机+加一点点优化,否则过不了,具体见代码,就一行代码。
传送门

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=1000005;//字符串长度
struct ac_auto{
    const static int letter_size = 26;
    const static int trie_size = (1e6+50);
    int tot , root ,fail[trie_size],end[trie_size],next[trie_size][letter_size];
    int newnode()
    {
        end[tot]=0;
        memset(next[tot],-1,sizeof next[tot]);
        return tot++;
    }
    void init()
    {
        tot=0;
        root=newnode();
    }
    int getidx(char x){
        return x-'a';
    }
    void insert(char ss[])//建字典树
    {
        int len=strlen(ss);
        int now=root;
        for(int i=0;i<len;i++)
        {
            int id=getidx(ss[i]);
            if(next[now][id]==-1)
            {
                next[now][id]=newnode();
            }
            now=next[now][id];
        }
        end[now]++;
    }
    void getfail()//求出fail指针
    {
        queue<int>q;
        fail[root]=root;
        for(int i=0;i<letter_size;i++)//初始化
        {
            if(next[root][i]!=-1)
            {
                fail[next[root][i]]=root;
                q.push(next[root][i]);
            }
            else
            {
                next[root][i]=root;
            }
        }
        while(!q.empty())
        {
            int now=q.front();q.pop();
            for(int i=0;i<letter_size;i++)
            {
                if(next[now][i]!=-1)
                {
                    fail[next[now][i]]=next[fail[now]][i];//相当于next[now][i]新加入一个分支,用fail来维护这条分支的链接。
                    q.push(next[now][i]);
                }
                else next[now][i]=next[fail[now]][i];//这里因为next[now][i]没有后继,直接指向另一条分支
            }
        }
    }
    int match(char ss[])
    {
        int sum=0,now=root;
        int len=strlen(ss);
        for(int i=0;i<len;i++)
        {
            int id=getidx(ss[i]);
            int tmp=now=next[now][id];
            while(tmp)
            {
                //不优化
                //if(end[tmp]!=-1)
                //res += end[tmp];
                //优化
                if(end[tmp]==-1)break;//移边删除小优化
                sum+=end[tmp];
                end[tmp]=-1;
                tmp=fail[tmp];
            }
        }
        return sum;
    }

};
ac_auto ac;
char ss[maxn];
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        ac.init();
        for(int i=0;i<n;i++)
        {
            scanf("%s",ss);
            ac.insert(ss);
        }
        ac.getfail();
        scanf("%s",ss);
        printf("%d\n",ac.match(ss));
    }
    return 0;
}
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,以下是 AC 自动机模板代码: ```C++ #include <iostream> #include <cstring> #include <queue> using namespace std; const int MAXN = 100010; const int MAXM = 26; struct TrieNode { int fail; int next[MAXM]; int cnt; } trie[MAXN]; int nodeCnt; void insertTrie(char* str) { int cur = 0; for (int i = 0; str[i]; i++) { int t = str[i] - 'a'; if (!trie[cur].next[t]) { trie[cur].next[t] = ++nodeCnt; } cur = trie[cur].next[t]; } trie[cur].cnt++; } void buildAC() { queue<int> q; for (int i = 0; i < MAXM; i++) { if (trie[0].next[i]) { q.push(trie[0].next[i]); } } while (!q.empty()) { int cur = q.front(); q.pop(); for (int i = 0; i < MAXM; i++) { if (trie[cur].next[i]) { int fail = trie[cur].fail; while (fail && !trie[fail].next[i]) { fail = trie[fail].fail; } trie[trie[cur].next[i]].fail = trie[fail].next[i]; q.push(trie[cur].next[i]); } } } } int queryAC(char* str) { int cur = 0, res = 0; for (int i = 0; str[i]; i++) { int t = str[i] - 'a'; while (cur && !trie[cur].next[t]) { cur = trie[cur].fail; } cur = trie[cur].next[t]; int tmp = cur; while (tmp && trie[tmp].cnt != -1) { res += trie[tmp].cnt; trie[tmp].cnt = -1; tmp = trie[tmp].fail; } } return res; } int main() { int n; char str[MAXN]; cin >> n; for (int i = 0; i < n; i++) { cin >> str; insertTrie(str); } buildAC(); cin >> str; cout << queryAC(str) << endl; return 0; } ``` 这个模板实现了 AC 自动机的构建和查询功能,可以在其中插入多个字符串,然后查询一个字符串中有多少个字符串在之前插入的字符串中出现过。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值