LA 4670 (AC自动机 模板题) Dominating Patterns

AC自动机大名叫Aho-Corasick Automata，不知道的还以为是能自动AC的呢，虽然它确实能帮你AC一些题目。=_=||

AC自动机看了好几天了，作用就是多个模式串在文本串上的匹配。

因为有多个模式串构成了一颗Tire树，不能像以前一样线性递推失配函数f了，于是改成了BFS求失配函数。

白书上那个last数组(后缀链接)的含义就是：在Tire树的某个分支上虽然没有遇到单词节点，但是某个单词可能是已经匹配上的字串的后缀。

举个栗子：

有两个模式串：aaabbb, ab

现在已经匹配了aaab，现在的位置正在Tire树上aaabbb的分支上，明显没有遇到单词节点，但是注意到ab是aaab的后缀，也就是说我们虽然没有匹配到第一个模式串，但是找到第二个模式串ab了。

这就是last数组的作用。

  1 #include <cstdio>

  2 #include <string>

  3 #include <cstring>

  4 #include <queue>

  5 #include <map>

  6 using namespace std;

  7 

  8 const int SIGMA_SIZE = 26;

  9 const int MAXNODE = 11000;

 10 const int MAXS = 150 + 10;

 11 

 12 map<string, int> ms;

 13 

 14 struct AhoCorasickAutomata

 15 {

 16     int ch[MAXNODE][SIGMA_SIZE];

 17     int f[MAXNODE];    //失配函数

 18     int val[MAXNODE];

 19     int last[MAXNODE];

 20     int cnt[MAXS];

 21     int sz;

 22 

 23     void init()

 24     {

 25         memset(ch[0], 0, sizeof(ch[0]));

 26         memset(cnt, 0, sizeof(cnt));

 27         ms.clear();

 28         sz = 1;

 29     }

 30 

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

 32 

 33     //插入字符串s, v > 0

 34     void insert(char* s, int v)

 35     {

 36         int u = 0, n = strlen(s);

 37         for(int i = 0;  i < n; i++)

 38         {

 39             int c = idx(s[i]);

 40             if(!ch[u][c])

 41             {

 42                 memset(ch[sz], 0, sizeof(ch[sz]));

 43                 val[sz] = 0;

 44                 ch[u][c] = sz++;

 45             }

 46 

 47             u = ch[u][c];

 48         }

 49         val[u] = v;

 50         ms[string(s)] = v;

 51     }

 52 

 53     //统计每个字串出现的次数

 54     void count(int j)

 55     {

 56         if(j)

 57         {

 58             cnt[val[j]]++;

 59             count(last[j]);

 60         }

 61     }

 62 

 63     //在T中查找模板

 64     void find(char* T)

 65     {

 66         int j = 0, n = strlen(T);

 67         for(int i = 0; i < n; i++)

 68         {

 69             int c = idx(T[i]);

 70             while(j && !ch[j][c]) j = f[j];

 71             j = ch[j][c];

 72             if(val[j]) count(j);

 73             else if(last[j]) count(last[j]);

 74         }

 75     }

 76 

 77     //计算失配函数

 78     void getFail()

 79     {

 80         queue<int> q;

 81         f[0] = 0;

 82         for(int i = 0; i < SIGMA_SIZE; i++)

 83         {

 84             int u = ch[0][i];

 85             if(u) { last[u] = 0; f[u] = 0; q.push(u); }

 86         }

 87         while(!q.empty())

 88         {

 89             int r = q.front(); q.pop();

 90             for(int c = 0; c < SIGMA_SIZE; c++)

 91             {

 92                 int u = ch[r][c];

 93                 if(!u) continue;

 94                 q.push(u);

 95                 int v = f[r];

 96                 while(v && !ch[v][c]) v = f[v];

 97                 f[u] = ch[v][c];

 98                 last[u] = val[f[u]] ? f[u] : last[f[u]];

 99             }

100         }

101     }

102 }ac;

103 

104 const int maxn = 1000000 + 10;

105 char T[maxn], P[160][80];

106 

107 int main()

108 {

109     //freopen("in.txt", "r", stdin);

110 

111     int n;

112     while(scanf("%d", &n) == 1 && n)

113     {

114         ac.init();

115         for(int i = 1; i <= n; i++) { scanf("%s", P[i]); ac.insert(P[i], i); }

116         scanf("%s", T);

117         ac.getFail();

118         ac.find(T);

119         int M = -1;

120         for(int i = 1; i <= n; i++) if(ac.cnt[i] > M) M = ac.cnt[i];

121         printf("%d\n", M);

122         for(int i = 1; i <= n; i++)

123             if(ac.cnt[ms[string(P[i])]] == M)

124                 printf("%s\n", P[i]);

125     }

126 

127     return 0;

128 }

代码君