hdu 2825 Wireless Password

题目链接：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();
    }
}