hdu 2825 Wireless Password

제목 링크:http://acm.hdu.edu.cn/showproblem.php?pid=2825
제목: m (m < = 10) 개의 문자열 (길이 < = 10) 이 있 는 집합 을 지정 합 니 다. 그 중에서 k 개의 문자열 (k < = m) 보다 적지 않 은 것 을 취하 고 n 의 문자열 로 조합 할 수 있 습 니 다. 이 문자열 은 서로 중첩 되 거나 중복 되 며 몇 가지 조합 이 있 는 지 물 어 볼 수 있 습 니 다.
제목 사고: ac 자동 동기 dp, 그러나 이 문 제 는 상수 적 으로 최적화 되 었 습 니 다. 그러나 최근 에 인품 이 좋 지 않 습 니 다. 각종 최적화 후에 효과 가 좋 지 않 습 니 다. 마지막 으로 효과 가 좋 지 않다 고 생각 했 던 최 적 화 를 해 보 았 습 니 다. 그 결과 ac 가 되 었 습 니 다. 어 지 럽 습 니 다. 그리고 원래 의 최적화 까지 해서 1 등 을 했 습 니 다.
#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
",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(); } }

좋은 웹페이지 즐겨찾기