题目链接:这里写链接内容
题目描述:给定n、m、k,其中n为密码的长度,m为已知串的数量,k为密码中应该包含的已知串的数量,问密码有多少种可能?
解题思路:AC自动机+状压dp
状态:dp[i][j][k]
其中i为密码的长度,j为当前在AC自动机中所处的状态,k为包含的已知串的信息,k转化为二进制,如果第x位为1则说明包含第x个已知串
状态转移方程:
dp[i+1][next1[j][u]][k|end1[next1[j][u]]]+=dp[i][j][k];
代码:
#include <cstdio>
#include <cstring>
#include <queue>
#define MOD 20090717
using namespace std;
int _hash[1024];
int dp[26][110][1024];
struct Trie
{
int next1[110][26],fail[110];
int end1[110];
int root,L;
int newnode()
{
for(int i=0; i<26; ++i) next1[L][i]=-1;
end1[L++]=0;
return L-1;
}
void init()
{
L=0;
root=newnode();
}
void insertnode(char* str,int id)
{
int len=strlen(str);
int now=root;
for(int i=0; i<len; ++i)
{
int to=str[i]-'a';
if(next1[now][to]==-1) next1[now][to]=newnode();
now=next1[now][to];
}
end1[now]|=(1<<id);
}
void build()
{
fail[root]=root;
queue<int> q;
for(int i=0; i<26; ++i)
{
if(next1[root][i]==-1)
next1[root][i]=root;
else
{
fail[next1[root][i]]=root;
q.push(next1[root][i]);
}
}
while(!q.empty())
{
int now=q.front();
q.pop();
if(end1[fail[now]]) end1[now]=(end1[now]|end1[fail[now]]);
for(int i=0; i<26; ++i)
{
if(next1[now][i]==-1)
next1[now][i]=next1[fail[now]][i];
else
{
fail[next1[now][i]]=next1[fail[now]][i];
q.push(next1[now][i]);
}
}
}
}
int solve(int n,int kk)
{
memset(dp,0,sizeof(dp));
dp[0][0][0]=1;
for(int i=0; i<n; ++i) for(int j=0; j<L; ++j)for(int k=0; k<1024; ++k)
if(dp[i][j][k]) for(int u=0;u<26;u++){
dp[i+1][next1[j][u]][k|end1[next1[j][u]]]+=dp[i][j][k];
dp[i+1][next1[j][u]][k|end1[next1[j][u]]]%=MOD;
}
int ans=0;
for(int i=0; i<L; ++i) for(int j=0;j<1024;++j)
if(_hash[j]>=kk)
{
ans+=dp[n][i][j];
ans%=MOD;
}
return ans;
}
};
int lowbit(int x)
{
int t=0;
while(x)
{
x=x&(x-1);
t++;
}
return t;
}
Trie ac;
char st[20];
int n,m,k;
int main()
{
for(int i=0; i<1024; ++i)
_hash[i]=lowbit(i);
while(scanf("%d%d%d",&n,&m,&k)!=EOF&&(n!=0||m!=0||k!=0))
{
ac.init();
for(int i=0; i<m; ++i)
{
scanf("%s",st);
ac.insertnode(st,i);
};
ac.build();
printf("%d\n",ac.solve(n,k));
}
return 0;
}