题目大意
有 n n n 个串, s 1 , s 2 , … s n s_1,s_2,\dots s_n s1,s2,…sn。
试构造一个长度为 k k k( k k k 是给定的)的串 x x x,使得对于 ∀ 1 ≤ i ≤ n , s i \forall 1 \leq i \leq n,s_i ∀1≤i≤n,si 在 x x x 中的出现次数之和最大。
输出这个最大值。
串只包含 A
,B
,C
三个字符。
解题思路
前缀知识:AC
自动机
考虑对
s
i
s_i
si 建 AC
自动机。
然后 dp
。
记
d
p
[
i
]
[
u
]
dp[i][u]
dp[i][u] 表示为长度为
i
i
i 的字符串,且当前已计算的节点是 Trie
上的编号为
u
u
u 的节点的最大得分。
记
t
a
g
[
k
]
tag[k]
tag[k] 表示在 Trie
以点
k
k
k 为结尾的字符串的个数。
然后将 t a g tag tag 下传。
则转移方程为:
for (int i = 0; i < m; ++i)
for (int j = 0; j <= cnt; ++j)
for (int k = 0; k < 3; ++k)
dp[i + 1][tr[j][k]] = max(dp[i + 1][tr[j][k]], dp[i][j] + tag[tr[j][k]]);
注意我这里是以
0
0
0 为 Trie
的根节点。
m m m 是读入的 k k k。
c
n
t
cnt
cnt 是 Trie
上的节点个数。
答案为:
for (int i = 0; i <= cnt; ++i)
ans = max(ans, dp[m][i]);
AC CODE
#include <bits/stdc++.h>
using namespace std;
#define int long long
inline int read()
{
char tr = getchar();
int x = 0;
while (tr < '0' || tr > '9')
tr = getchar();
while (tr >= '0' && tr <= '9')
x = x * 10 + tr - '0', tr = getchar();
return x;
}
#define _ (int)1e5 + 7
int n, m, cnt, tr[_][3], dp[2000][500], fail[_];
int tag[_];
char str[_];
void insert(char *str)
{
int now = 0;
int len = strlen(str);
for (int i = 0; i < len; ++i)
{
int p = str[i] - 'A';
if (!tr[now][p])
tr[now][p] = ++cnt;
now = tr[now][p];
}
tag[now]++;
}
void get_fail()
{
queue<int> q;
for (int i = 0; i < 3; ++i)
if (tr[0][i])
{
fail[tr[0][i]] = 0;
q.push(tr[0][i]);
}
while (!q.empty())
{
int u = q.front();
q.pop();
for (int i = 0; i < 3; ++i)
if (tr[u][i])
{
fail[tr[u][i]] = tr[fail[u]][i];
q.push(tr[u][i]);
}
else
tr[u][i] = tr[fail[u]][i];
tag[u] += tag[fail[u]];
}
}
signed main()
{
n = read();
m = read();
for (int i = 1; i <= n; ++i)
{
scanf("%s", str);
insert(str);
}
get_fail();
int ans = 0;
for(int i = 0; i < m; ++i)
for(int j = 1; j <= cnt; ++j) dp[i][j] = INT_MIN;
for (int i = 0; i < m; ++i)
for (int j = 0; j <= cnt; ++j)
for (int k = 0; k < 3; ++k)
dp[i + 1][tr[j][k]] = max(dp[i + 1][tr[j][k]], dp[i][j] + tag[tr[j][k]]);
for (int i = 0; i <= cnt; ++i)
ans = max(ans, dp[m][i]);
printf("%lld\n", ans);
}
/*
20 1000
CACAACCCCBACA
ACCACAACC
ACAABCACACCACA
A
CCACAACCCCBACA
BCACACCACAAC
CBACAABCACACCA
CCACAACCCC
CACCACAACCCCBA
AABCACACCA
CCCCBACAAB
ACCACAACCCCBA
ACCACAACCCCBAC
C
CAACCCCBACAA
ACAACCCCBACAAB
CCACAACC
ACAABCACACCACAA
CCACAACCCCBACAA
ACAACCCCBACAABC
1795
*/