NYOJ 1085 数单词 （AC自己主动机模板题）

数单词

时间限制：1000 ms  |  内存限制：65535 KB

难度：4

描写叙述

为了可以顺利通过英语四六级考试，如今大家每天早上都会早起读英语。

LYH本来以为自己在6月份的考试中能够通过六级，但是没想到，成绩出来以后。竟然没有通过。所以他不得不付出很多其它的时间来学习英语。

要想通过六级。最主要的要求就是词汇量。为了可以更快的记住一些陌生单词。LYH有时会找一些英语文章来读。

今天早上。LYH又找了一篇文章。读之前。他突然萌生出一个想法：文章中哪些单词出现的次数最多呢？

输入

第一行输入一个整数T。表示有T组測试数据（1≤T≤200）。
对于每组測试数据。第一行输入一个整数n（1≤n≤150），表示LYH要查询的单词数量（有些单词可能会反复出现）。
接下来n行。每行输入一个单词，长度不大于100。
最后一行包括一个由小写字母组成的英语文章（字符串），长度不大于10^6。

输出

对于每组数据，第一行输出一个整数，表示单词出现的次数。
然后依照输入顺序，每行输出一个出现次数最多的单词。

假设有反复出现的单词。把它们所有输出。

例子输入

2
3
good
oo
one
goodafternooneveryone
1
to
welcometotopcoder

例子输出

2
oo
one
2
to

分析：这就是一个AC自己主动机模板题，要注意的是查询的单词中，一个单词可能会出现多次。这里要处理一下。

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <map>
#include <string>
#include <queue>
using namespace std;

#define SIGMA_SIZE 26 //文本串字符内容
#define MAXNODE 20000 //节点数量
#define TEXT_SIZE 1000005 //文本串长度
#define P_SIZE 100 //模式串长度
#define P_NUM 200 //模式串数量

map <string, int> mp;

struct AhoCorasickAutomata
{
    int cnt[P_NUM];
    int sz;
    int ch[MAXNODE][SIGMA_SIZE];
    int f[MAXNODE];
    int val[MAXNODE];
    int last[MAXNODE];

    void Init() {
        sz = 1;
        memset(ch[0],0,sizeof(ch[0]));
        memset(cnt,0,sizeof(cnt));
        mp.clear();
    }

    int idx(char c) {
        return c - 'a';
    }

    void Insert(char *s,int v) {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c]) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                val[sz] = 0;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
        mp[string(s)] = v;
    }

    void print(int j) {
        if(j) {
            cnt[val[j]]++;
            print(last[j]);
        }
    }

    void Find(char *T) {
        int n = strlen(T);
        int j = 0;
        for(int i = 0; i < n; i++) {
            int c = idx(T[i]);
            while(j && !ch[j][c]) j = f[j];
            j = ch[j][c];
            if(val[j]) print(j);
            else if(last[j]) print(last[j]);
        }
    }

    void Get_Fail() {
        queue<int> q;
        f[0] = 0;
        for(int c = 0; c<SIGMA_SIZE; c++) {
            int u = ch[0][c];
            if(u) {
                f[u] = 0;
                q.push(u);
                last[u] = 0;
            }
        }
        while(!q.empty()) {
            int r = q.front();
            q.pop();
            for(int c = 0; c<SIGMA_SIZE; c++) {
                int u = ch[r][c];
                if(!u) continue;
                q.push(u);
                int v = f[r];
                while(v && !ch[v][c]) v = f[v];
                f[u] = ch[v][c];
                last[u] = val[f[u]] ? f[u] : last[f[u]];
            }
        }
    }
};

char text[TEXT_SIZE];
char P[P_NUM][P_SIZE];
AhoCorasickAutomata ac;
int n, T;

int main() {
    scanf("%d", &T);
    int cas = 0;
    while(T--) {
        scanf("%d", &n);
        ac.Init();
        for(int i = 1; i <= n; i++) {
            scanf("%s", P[i]);
            ac.Insert(P[i], i);
        }
        ac.Get_Fail();
        scanf("%s", text);
        ac.Find(text);
        int Max_cnt = -1;
        for(int i = 1; i <= n; i++)
            if(ac.cnt[i] > Max_cnt)
                Max_cnt = ac.cnt[i];
        printf("%d
", Max_cnt);
        for(int i = 1; i <= n; i++)
            if(ac.cnt[mp[string(P[i])]] == Max_cnt)
                printf("%s
", P[i]);
    }
    return 0;
}