题目:http://acm.hdu.edu.cn/showproblem.php?pid=2825
题意:给定m个模式串,问有多少个长为n的文本串至少包含K种模式串。
分析:首先将模式串建ac自动机,然后在求fail指针的时候把到达某个节点需要加的贡献求出来(由于最多只有10种模式串,可以将模式串种数压缩到一个int型)。然后记忆化搜索,dp[len][cur][num],len代表当前字符串长度,cur代表对应在ac机上的位置,num代表已经选的模式串的种数,dp[len][cur][num]表示方案数。ps:dp数组清零用memset超时,测试数据组数比较多吧。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
typedef long long LL;
const LL mod = 20090717;
const int maxn = 108;
const int kd = 26;
int N;
struct trie
{
int son[maxn][kd],fail[maxn],end[maxn];
int cnt,root;
int L,K;
LL dp[27][maxn][1200];
int newnode()
{
fill(son[cnt],son[cnt]+kd,-1);
end[cnt]=0;
return cnt++;
}
void Init()
{
cnt=0;
root=newnode();
}
void Insert(char str[],int id)
{
int index,i,now=root;
for(i=0;str[i];i++)
{
index=str[i]-'a';
if(son[now][index]==-1)
son[now][index]=newnode();
now=son[now][index];
}
end[now]|=(1<<id);
}
void findfail()
{
queue <int > q;
fail[root]=root;
int i,temp;
for(i=0;i<kd;i++)
if(son[root][i]==-1)
son[root][i]=root;
else
{
fail[son[root][i]]=root;
q.push(son[root][i]);
}
while(!q.empty())
{
temp=q.front();
q.pop();
end[temp]|=end[fail[temp]];
for(i=0;i<kd;i++)
if(son[temp][i]==-1)
son[temp][i]=son[fail[temp]][i];
else
{
fail[son[temp][i]]=son[fail[temp]][i];
q.push(son[temp][i]);
}
}
}
LL DFS(int len,int cur,int num)
{
if(len==L)
{
int i,c;
for(i=0,c=0;i<10;i++)
if(num&(1<<i))
c++;
return c>=K;
}
if(dp[len][cur][num]!=-1)
return dp[len][cur][num];
LL ret=0;
for(int i=0;i<kd;i++)
ret+=DFS(len+1,son[cur][i],num|end[son[cur][i]])%mod;
return dp[len][cur][num]=ret%mod;
}
void solve()
{
for(int i=0;i<=L;i++)
for(int j=0;j<=cnt;j++)
for(int k=0;k<=(1<<N);k++)
dp[i][j][k]=-1;
LL ans=DFS(0,root,0);
printf("%I64d\n",ans);
}
}ac;
char s[maxn];
int main()
{
while(scanf("%d%d%d",&ac.L,&N,&ac.K),(ac.L || N || ac.K))
{
ac.Init();
for(int i=0;i<N;i++)
{
scanf("%s",s);
ac.Insert(s,i);
}
ac.findfail();
ac.solve();
}
return 0;
}