hdu 3341 Lost's revenge

題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=3341

題目大意:給出一些病毒串,一個原串,問怎麼調整原串的順序使得跟最多的病毒串匹配。

題目思路:用dp[a][b][c][d][e]表示分別用了a,b,c,d個鹼基,到達e結點的最值。開始加了一個總長度,後來想到既然有總長度,則有一維可以去,但是還是超時,哎,居然沒有想到不用總長度直接用4種鹼基的長度來表示狀態,這樣就又可以降低常數了,居然一下就刷到第一了,嘿嘿。。

#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\n",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);

    }

}


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章