欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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