AC自动机模板题 P2017

Description

给出由若干单词组成的字典（可能有相同的单词），再给出一篇文章（一个字符串），请计算有多少个不同的单词在字符串中出现。

---

Input

第1行一个正整数N。表示字典中单词的数目。接下来的N行，每行一个字符串（长度不超过50），表示一个单词。最后一行为一个长字符串（长度不超过100000）。注意：单词和字符串均由小写字母构成。

---

Output

一行一个整数，表示长字符串中包含单词的数量。

---

Hint

N<=40000

---

Solution

AC自动机其实就是kmp+trie树（但实际上我觉得并没有kmp代码基本上还是按照trie树来的。。。），然后大概就是先把每个模式串按照前缀存进trie树，然后，，，，setnext数组，我是把next放进结构体里面存的所以还行，然后最后dofind的过程就是用主串来匹配模式串构成的trie树，然后就完了，，，挺好理解的就是代码写着令人智熄。。

注意事项： 1、newp要初始化成1因为是从1结点开始存的，0结点在setnext的过程中要用。不初始化会爆。。。 2、然后为了防止比如 abc ab 这种情况的时候如果就这样读的话读abc这个串的时候会读一个b标记的end_line和c标记的end_line，就已经把ab这个前缀读了，再读ab的话会重复，所以在cnt+=tree[k].end_line之后就把end_line给赋值为false就会防止读重复的情况了。 3、我是傻逼。。。数组要开到两百万我又只开了四万结果RE了一半。。有毒啊拉低我正确率我嚓本来我能一遍AC的。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define maxm 27
#define maxn 2000005
using namespace std;
struct Trie{
    int ch[maxm];
    bool end_line;
    int next;
}tree[maxn];
int newp=1,n,cnt;
char tmp[maxn],S[maxn];
void insert(char *s){
    int x=1,lenofS=strlen(s+1);
    for(int i=1;i<=lenofS;i++){
        int y=s[i]-'a';
        if(tree[x].ch[y]==0){
            tree[x].ch[y]=++newp;
        }
        x=tree[x].ch[y];
    }
    tree[x].end_line=true;
}
void setnext(){
    for(int i=0;i<26;i++){
        tree[0].ch[i]=1;
    }
    queue<int>q;
    q.push(1);
    tree[1].next=0;
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(int i=0;i<26;i++){
            if(tree[u].ch[i]==0){
                tree[u].ch[i]=tree[tree[u].next].ch[i];
            }
            else{
                q.push(tree[u].ch[i]);
                int v=tree[u].next;
                tree[tree[u].ch[i]].next=tree[v].ch[i];
            }
        }
    }
}
void dofindkmp(){
    int u=1,lenofS=strlen(S+1);
    for(int i=1;i<=lenofS;i++){
        int c=S[i]-'a';
        int k=tree[u].ch[c];
        while(k>1){
            cnt+=tree[k].end_line;
            tree[k].end_line=false;
            k=tree[k].next;
        }
        u=tree[u].ch[c];
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",tmp+1);
        insert(tmp);
    }
    scanf("%s",S+1);
    setnext();
    dofindkmp();
    printf("%d\n",cnt);
    return 0;
}

转载于:https://www.cnblogs.com/virtual-north-Illya/p/10045139.html