hdu 2825 Wireless Password
程序员文章站
2022-03-23 11:37:31
...
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2825
题目大意:给定1个有m(m<=10)个字符串(长度<=10)的集合,从其中取不少于k个串(k<=m),拼成一个长为n的串,这个串可以互相重叠或重复,问有多少种组合。
题目思路:ac自动机dp,不过这个题考常量优化,不过最近人品不好啊,各种优化后效果都不好,最后试了一下原以为效果不好的优化,结果ac了,晕,加上原先的优化就跑到了第一。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
#define mod 20090717
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int q[12*12],cnt,num[2200];
int b[120][120],mp[120][30];
int dp[26][1025][120];
int n,m,k;
struct node
{
int cnt,fail;
int next[26];
void init()
{
cnt=fail=0;
memset(next,0,sizeof(next));
}
}tri[12*12];
void init()
{
for(int i=0;i<(1<<10);i++)
{
for(int j=0;j<10;j++)
{
if((i&(1<<j))!=0)
{
num[i]++;
}
}
}
}
void insert(char *s,int id)
{
int i,p,x;
p=0;
for(i=0;s[i];i++)
{
x=s[i]-'a';
if(!tri[p].next[x])
{
tri[++cnt].init();
tri[p].next[x]=cnt;
}
p=tri[p].next[x];
}
tri[p].cnt|=(1<<id);
}
void bfs()
{
int i,p=0,suf,head=0,tail=0;
for(i=0;i<26;i++)
{
if(tri[0].next[i])
{
q[tail++]=tri[0].next[i];
tri[q[tail-1]].fail=0;
}
}
while(head<tail)
{
p=q[head++];suf=tri[p].fail;
tri[p].cnt|=tri[suf].cnt;
for(i=0;i<26;i++)
{
if(tri[p].next[i])
{
q[tail++]=tri[p].next[i];
tri[q[tail-1]].fail=tri[suf].next[i];
}
else
tri[p].next[i]=tri[suf].next[i];
}
}
memset(b,0,sizeof(b));
for(i=0;i<=cnt;i++)
{
for(int j=0;j<26;j++)
{
int tmp=tri[i].next[j];
b[i][tmp]++;
}
}
memset(mp,0,sizeof(mp));
for(i=0;i<=cnt;i++)
{
for(int j=0;j<=cnt;j++)
{
if(b[i][j])
{
mp[i][0]++;
mp[i][mp[i][0]]=j;
}
}
}
}
void solve()
{
int i,j,l,g,tmp;
int ans=0;
for(i=0;i<=n;i++)
for(j=0;j<(1<<m);j++)
for(l=0;l<=cnt;l++)
dp[i][j][l]=0;
dp[0][0][0]=1;
if(k==0) ans=1;
for(i=0;i<n;i++)
{
ans*=26;
ans%=mod;
for(j=0;j<(1<<m);j++)
{
if(num[j]>=k)
continue;
for(l=0;l<=cnt;l++)
{
if(!dp[i][j][l])
continue;
for(g=1;g<=mp[l][0];g++)
{
tmp=mp[l][g];
if(num[j|tri[tmp].cnt]<k)
{
dp[i+1][j|tri[tmp].cnt][tmp]+=b[l][tmp]*dp[i][j][l];
dp[i+1][j|tri[tmp].cnt][tmp]%=mod;
continue;
}
ans+=b[l][tmp]*dp[i][j][l];
ans%=mod;
}
}
}
}
printf("%d\n",ans%mod);
}
int main()
{
init();
int i;
char str[60];
while(scanf("%d%d%d",&n,&m,&k),n|m|k)
{
cnt=0;tri[0].init();
for(i=0;i<m;i++)
{
scanf("%s",str);
insert(str,i);
}
bfs();
solve();
}
}
上一篇: iOS Dark Mode快速适配