BZOJ1030 JSOI2007 文本生成器 题解&代码

题意：给出n个匹配串，询问：对于长度为m的串，有多少个串至少包含一个匹配串（答案对10007取模）

题解：

“至少包含一个匹配串的长度为m的串”，那么很容易转化为“所有串除去不包含任何匹配串的长度为m的串”

然后就是喜闻乐见的AC自动机上的dp了，dp方程显然是dp[i][j]表示长度为i的串匹配到j位时有多少不包含任何匹配串

有：dp[i][ch[j][k]]+=dp[i-1][j]

即孩子节点一定由有效父亲节点推知（节点本身flag指针就是true的当然是0）

/**************************************************************
Problem: 1030
User: Rainbow6174
Language: C++
Result: Accepted
Time:76 ms
Memory:4396 kb
****************************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#define MOD 10007
using namespace std;
const int maxt = 6005;
int n, m, tot, ans = 1, temp, ch[maxt][26], fail[maxt], q[maxt], dp[105][maxt];
char s[105];
bool flag[maxt];
void newnode(int x, int v)
{
ch[x][v]=++tot;
}
void addtrie(char s[])
{
int x = 0, p = 0, len = strlen(s);
while( p < len )
{
temp = s[p]-'A';
if ( !ch[x][temp] ) newnode(x, temp);
x = ch[x][temp];
p++;
}
flag[x] = 1;
}
void AC(void)
{
int h = 0, t = 0;
for(int i = 0; i < 26; i++)
if(ch[0][i]) q[t++] = ch[0][i];
while( h < t )
{
temp = q[h++];
for(int i = 0; i < 26; i++)
if( !ch[temp][i] ) ch[temp][i] = ch[fail[temp]][i];
else fail[ch[temp][i]] = ch[fail[temp]][i],
flag[ch[temp][i]] |= flag[fail[ch[temp][i]]],
q[t++] = ch[temp][i];
}
}
int main(void)
{
scanf("%d%d", &n, &m);
for(int i = 0; i < n; i++)
{
scanf("%s", s);
addtrie(s);
}
AC();
dp[0][0] = 1;
for(int i = 1; i <= m; i++)
{
for(int j = 0; j <= tot; j++)
{
if(flag[j] || !dp[i-1][j]) continue;
for(int k = 0; k < 26; k++)
{
dp[i][ch[j][k]] += dp[i-1][j];
dp[i][ch[j][k]] %= MOD;
}
}
ans*=26;
ans%=MOD;
}
//cout<<ans<<endl;
for(int i = 0; i <= tot; i++)
{
//cout<<i<<' '<<flag[i]<<endl;
if(!flag[i])
{
//cout<<dp[m][i]<<endl;
ans -= dp[m][i];
ans += MOD;
ans %= MOD;
}
}
printf("%d\n",ans);
return 0;
}