HDU-2222-Keywords Search(AC自动机模板题)

题目链接

问题描述:给你n个模式串，再给你一个目标串，问n个模式串中有多少串存在在目标串中。

思路:AC自动机模板题

#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>#include<iostream>#include<time.h>#include<set>#include<stack>#include<vector>#include<map>#include<queue>#define pi acos(-1)#define maxn 111111#define maxm 1111111#define INF 0x3F3F3F3F#define eps 1e-8#define pb push_back#define mem(a) memset(a,0,sizeof a)using namespace std;const long long mod = 1000000007;/**lyc**/char str[maxn][77];   ///模式串int n;char a[maxm];#define STAUTS_NUM 500000struct trie {    ///next,fail,end都是基于第几个状态 状态就是一个字符串顺序    int next[STAUTS_NUM][26], ///next数组表示 next[i][j]表示第i个状态后连j字符会变成什么状态，                        ///注意最开始初始化的状态，在后面会被不断的优化    fail[STAUTS_NUM],         ///失配数组，表示应该回到一个之前已匹配部分的最长后缀那里开始重新匹配。类似kmp    end[STAUTS_NUM];          ///实际上就是表示哪些序列是模式串中有的，1,0，表示  //  int next2[maxn][26]; /**这个数组是我这个程序中另外设的,用来储存最原始的next数组，有特殊用**/    int num[maxn];///每个单词出现的次数    int f[STAUTS_NUM];///每个节点对应的单词编号    int root,  ///根节点状态，默认是0    cnt;       ///状态数    int new_node () {  ///对于一个没有的状态，对他新建一个节点        memset (next[cnt], -1, sizeof next[cnt]);        end[cnt++] = 0;        return cnt-1;    }    void init () {    ///初始化这棵字典树        cnt = 0;        root = new_node ();        memset (num, 0, sizeof num);        memset (f, -1, sizeof f);    }    void insert (char *buf, int pos) { ///字典树插入一个模式串        int len = strlen (buf);        int now = root;        for (int i = 0; i < len; i++) {            int id = buf[i] - 'a';            if (next[now][id] == -1) {  ///没有这个状态就新建一个节点                next[now][id] = new_node ();            }            now = next[now][id];        }        end[now]++;        f[now] = pos;   ///标记一下这个状态对应第几个字符串    }    void build () {   ///构建fail数组        /**这里备份一下next数组        for(int i = 0; i < cnt; i++) {            for(int j = 0; j < 26; j++) {                next2[i][j] = next[i][j];            }        }        这里备份一下next数组**/        queue <int> q;        fail[root] = root;        for (int i = 0; i < 26; i++) {            if (next[root][i] == -1) {                next[root][i] = root;            }            else {                fail[next[root][i]] = root;                q.push (next[root][i]);            }        }        while (!q.empty ()) {            int now = q.front (); q.pop ();            for (int i = 0; i < 26; i++) {                if (next[now][i] == -1) {                    next[now][i] = next[fail[now]][i];                }                else {                    fail[next[now][i]] = next[fail[now]][i];                    q.push (next[now][i]);                }            }        }    }    int query (char *buf) {        int len = strlen (buf);        int now = root;        int res = 0;        for(int i = 0; i < len; i++) {            int id = buf[i]-'a';            now = next[now][id];            int tmp = now;            while (tmp != root) {                if (end[tmp]) {                    res += end[tmp];                    end[tmp] = 0; ///问题求的是多少把钥匙能够匹配，                                  ///一种钥匙匹配后就要把这种状态给消掉，省的后面继续算。                }                tmp = fail[tmp];//沿着失配边走            }        }        printf("%ld\n", res);    }}ac;int main() {    int t;    scanf("%d", &t);    while(t--) {        scanf("%d", &n);        ac.init();        for(int i = 1; i <= n; i++) {            scanf("%s", str[i]);            ac.insert(str[i], i);        }        ac.build();        scanf("%s", a);        ac.query(a);    }    return 0;}