luogu P3796模板AC自动机（加强版）

Posted 2021-01-11 mrclr

嘟嘟嘟

 

这个和某谷的AC自动机模板简单版差不多。

但还是要注意几点的：

1.这个是统计出现次数，而不是是否出现，所以在查询的时候加上这个节点的val后，不能把val标记为-1。那么也就可以说查询的时间复杂度能比简单版的稍微第一慢一点。

2.考虑k个一样的模式串：刚开始我想的是每一个节点开一个vector，记录这里是第几个模式串。但其实没有这个必要，对于相同的模式串，我们只用记录任意一个就行，反而在出现次数上要都加上。因为如果主串中存在这些相同的模式串，那么出现次数应该是出现次数 * k。输出的时候如果是这些串最多，那么都应该把这些输出。

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<algorithm>
  4 #include<cmath>
  5 #include<cstring>
  6 #include<cstdlib>
  7 #include<cctype>
  8 #include<stack>
  9 #include<queue>
 10 #include<vector>
 11 using namespace std;
 12 #define enter puts("")
 13 #define space putchar(‘ ‘)
 14 #define Mem(a, x) memset(a, x, sizeof(a))
 15 #define rg register
 16 typedef long long ll;
 17 typedef double db;
 18 const int INF = 0x3f3f3f3f;
 19 const db eps = 1e-8;
 20 const int maxn = 1e6 + 5;
 21 const int maxm = 1.05e4 + 5;
 22 inline ll read()
 23 {
 24   ll ans = 0;
 25   char ch = getchar(), las = ‘ ‘;
 26   while(!isdigit(ch)) las = ch, ch = getchar();
 27   while(isdigit(ch)) ans = ans * 10 + ch - ‘0‘, ch = getchar();
 28   if(las == ‘-‘) ans = -ans;
 29   return ans;
 30 }
 31 inline void write(ll x)
 32 {
 33   if(x < 0) putchar(‘-‘), x = -x;
 34   if(x >= 10) write(x / 10);
 35   putchar(x % 10 + ‘0‘);
 36 }
 37 
 38 int n;
 39 char s[maxn], ss[155][75];
 40 
 41 int sum[155];
 42 int ch[maxm][26], val[maxm], pos[maxm], f[maxm], cnt = 0;
 43 int getnum(char c)
 44 {
 45   return c - ‘a‘;
 46 }
 47 void insert(int id, char *s)
 48 {
 49   int m = strlen(s);
 50   int now = 0;
 51   for(int i = 0; i < m; ++i)
 52     {
 53       int c = getnum(s[i]);
 54       if(!ch[now][c]) ch[now][c] = ++cnt;
 55       now = ch[now][c];
 56     }
 57   val[now]++; pos[now] = id;
 58 }
 59 void build()
 60 {
 61   queue<int> q;
 62   for(int i = 0; i < 26; ++i) if(ch[0][i]) q.push(ch[0][i]);
 63   while(!q.empty())
 64     {
 65       int now = q.front(); q.pop();
 66       for(int i = 0; i < 26; ++i)
 67     {
 68       if(ch[now][i]) f[ch[now][i]] = ch[f[now]][i], q.push(ch[now][i]);
 69       else ch[now][i] = ch[f[now]][i];
 70     }
 71     }
 72 }
 73 void query(char *s)
 74 {
 75   int m = strlen(s), now = 0;
 76   for(int i = 0; i < m; ++i)
 77     {
 78       int c = getnum(s[i]);
 79       now = ch[now][c];
 80       for(int j = now; j; j = f[j]) sum[pos[j]] += val[j];
 81     }
 82 }
 83 
 84 void init()
 85 {
 86   Mem(ch, 0); Mem(val, 0); Mem(pos, 0); Mem(f, 0);
 87   Mem(sum, 0);
 88   cnt = 0;
 89 }
 90 
 91 int main()
 92 {
 93   while(scanf("%d", &n) && n)
 94     {
 95       init();
 96       for(int i = 1; i <= n; ++i)
 97     {
 98       scanf("%s", ss[i]);
 99       insert(i, ss[i]);
100     }
101       build();
102       scanf("%s", s);
103       query(s);
104       int Max = -1;
105       for(int i = 1; i <= n; ++i) Max = max(Max, sum[i]);
106       write(Max); enter;
107       for(int i = 1; i <= n; ++i) if(sum[i] == Max) printf("%s
", ss[i]);
108     }
109   return 0;
110 }

View Code