hdu 3341 Lost's revenge

タイトルリンク:http://acm.hdu.edu.cn/showproblem.php?pid=3341
いくつかのウイルスの列を与えて、1つの原の列、どのように原の列の順序を調整して最も多いウイルスの列と一致させることを聞きます.
テーマ構想:dp[a][b][c][d][e]でそれぞれa,b,c,d個の塩基を用いてe接点に到達した最値を表す.最初は総长をプラスして、それから総长がある以上、一次元で行くことができると思っていましたが、タイムアウトしました.まあ、総长を使わずに直接4种类の塩基の长さで状态を表すとは思わなかった.これで定数を下げることができて、意外にもすぐに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 M 110
inline int max(int a,int b)
{
	return a>b?a:b;
}
int min(int a,int b)
{
	return a<b?a:b;
}
int num[4];
int cnt;
int dp[2][42][42][42][550];
char s[100],str[100];
int mp[1000],q[55*10];
struct node
{
    int cnt,fail;
    int next[4];
    void init()
    {
        cnt=fail=0;
        memset(next,0,sizeof(next));
    }
}tri[55*10];
void insert(char *s)
{
    int i,p,x;
    p=0;
    for(i=0;s[i];i++)
    {
        x=mp[s[i]];
        if(!tri[p].next[x])
        {
            tri[++cnt].init();
            tri[p].next[x]=cnt;
        }
        p=tri[p].next[x];//    cnt,          cnt;
    }
    tri[p].cnt++;
}
void bfs()
{
    int i,p,head,tail,suf;
    p=head=tail=0;
    for(i=0;i<4;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<4;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];
        }
    }
}
void set(int a)
{
     int j,b,c,d,e;
     for(b=0;b<=num[1];b++)
     for(c=0;c<=num[2];c++)
     for(d=0;d<=num[3];d++)
     for(e=0;e<=cnt;e++)
     {
        dp[a][b][c][d][e]=-1;
     }
}
void solve(int n)
{
    int tmp,i,j,a,b,c,d,e,pos,ans;
    tmp=0;
    set(0);
    dp[0][0][0][0][0]=0;
    for(a=0;a<=num[0];a++)
    {
        set(tmp^1);
        for(b=0;b<=num[1];b++)
        for(c=0;c<=num[2];c++)
        for(d=0;d<=num[3];d++)
        {
           for(e=0;e<=cnt;e++)
           {
               if(dp[tmp][b][c][d][e]==-1) continue;
                pos=tri[e].next[0];
                dp[tmp^1][b][c][d][pos]=max(dp[tmp^1][b][c][d][pos],dp[tmp][b][c][d][e]+tri[pos].cnt);
                pos=tri[e].next[1];
                dp[tmp][b+1][c][d][pos]=max(dp[tmp][b+1][c][d][pos],dp[tmp][b][c][d][e]+tri[pos].cnt);
                pos=tri[e].next[2];
                dp[tmp][b][c+1][d][pos]=max(dp[tmp][b][c+1][d][pos],dp[tmp][b][c][d][e]+tri[pos].cnt);
                pos=tri[e].next[3];
                dp[tmp][b][c][d+1][pos]=max(dp[tmp][b][c][d+1][pos],dp[tmp][b][c][d][e]+tri[pos].cnt);
           }
        }
        tmp^=1;
    }
    ans=0;
    for(i=0;i<=cnt;i++)
    {
        ans=max(ans,dp[tmp^1][num[1]][num[2]][num[3]][i]);
    }
    printf("%d
",ans);
}
int main()
{
    int n,i,len,count=1;
    mp['A']=0;mp['C']=1;mp['G']=2;mp['T']=3;
    while(scanf("%d",&n),n)
    {
        cnt=0;
        tri[0].init();
        for(i=0;i<n;i++)
        {
            scanf("%s",str);
            insert(str);
        }
        scanf("%s",s);
        len=strlen(s);
        memset(num,0,sizeof(num));
        for(i=0;i<len;i++)
        {
            num[mp[s[i]]]++;
        }
        bfs();
        printf("Case %d: ",count++);
        solve(len);

    }

}