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;
}