hdu4057 Rescue the Rabbit

地址：http://acm.hdu.edu.cn/showproblem.php?pid=4057

题目：

Rescue the Rabbit

Time Limit: 20000/10000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2233    Accepted Submission(s): 653

Problem Description

Dr. X is a biologist, who likes rabbits very much and can do everything for them. 2012 is coming, and Dr. X wants to take some rabbits to Noah's Ark, or there are no rabbits any more.

A rabbit's genes can be expressed as a string whose length is l (1 ≤ l ≤ 100) containing only 'A', 'G', 'T', 'C'. There is no doubt that Dr. X had a in-depth research on the rabbits' genes. He found that if a rabbit gene contained a particular gene segment, we could consider it as a good rabbit, or sometimes a bad rabbit. And we use a value W to measure this index.

We can make a example, if a rabbit has gene segment "ATG", its W would plus 4; and if has gene segment "TGC", its W plus -3. So if a rabbit's gene string is "ATGC", its W is 1 due to ATGC contains both "ATG"(+4) and "TGC"(-3). And if another rabbit's gene string is "ATGATG", its W is 4 due to one gene segment can be calculate only once.

Because there are enough rabbits on Earth before 2012, so we can assume we can get any genes with different structure. Now Dr. X want to find a rabbit whose gene has highest W value. There are so many different genes with length l, and Dr. X is not good at programming, can you help him to figure out the W value of the best rabbit.

 

Input

There are multiple test cases. For each case the first line is two integers n (1 ≤ n ≤ 10)，l (1 ≤ l ≤ 100), indicating the number of the particular gene segment and the length of rabbits' genes.

The next n lines each line contains a string DNAi and an integer wi (|wi| ≤ 100), indicating this gene segment and the value it can contribute to a rabbit's W.

 

Output

For each test case, output an integer indicating the W value of the best rabbit. If we found this value is negative, you should output "No Rabbit after 2012!".

 

Sample Input

2 4 ATG 4 TGC -3 1 6 TGC 4 4 1 A -1 T -2 G -3 C -4

 

Sample Output

4 4 No Rabbit after 2012!

Hint

case 1：we can find a rabbit whose gene string is ATGG(4), or ATGA(4) etc. case 2：we can find a rabbit whose gene string is TGCTGC(4), or TGCCCC(4) etc. case 3：any gene string whose length is 1 has a negative W.

 

Author

HONG, Qize

 

Source

2011 Asia Dalian Regional Contest

 

 思路：

　　dp[i][j][s]表示长度为i的串走到第j个节点时状态为s时是否可行。

　　如果用dp[i][j][s] 表示长度为i的串走到第j个节点时状态为s时的最小花费会Tle，因为常数会大十倍

　　而且需要状压，不然mle。

#include <queue>
#include <cstring>
#include <cstdio>
using namespace std;

const int INF=0x3f3f3f3f;
struct AC_auto
{
    const static int LetterSize = 4;
    const static int TrieSize = LetterSize * ( 1e3 + 50);

    int tot,root,fail[TrieSize],end[TrieSize],next[TrieSize][LetterSize];
    bool dp[2][TrieSize][1<<10];
    int newnode(void)
    {
        memset(next[tot],-1,sizeof(next[tot]));
        end[tot] = 0;
        return tot++;
    }

    void init(void)
    {
        tot = 0;
        root = newnode();
    }

    int getidx(char x)
    {
        if(x=='A') return 0;
        else if(x=='C') return 1;
        else if(x=='G') return 2;
        return 3;
    }

    void insert(char *ss,int x)
    {
        int len = strlen(ss);
        int now = root;
        for(int i = 0; i < len; i++)
        {
            int idx = getidx(ss[i]);
            if(next[now][idx] == -1)
                next[now][idx] = newnode();
            now = next[now][idx];
        }
        end[now]|=x;
    }

    void build(void)
    {
        queue<int>Q;
        fail[root] = root;
        for(int i = 0; i < LetterSize; i++)
            if(next[root][i] == -1)
                next[root][i] = root;
            else
                fail[next[root][i]] = root,Q.push(next[root][i]);
        while(Q.size())
        {
            int now = Q.front();Q.pop();
            for(int i = 0; i < LetterSize; i++)
            if(next[now][i] == -1)   next[now][i] = next[fail[now]][i];
            else
            {
                 fail[next[now][i]] = next[fail[now]][i];
                 end[next[now][i]]|=end[fail[next[now][i]]];
                 Q.push(next[now][i]);
            }
        }
    }

    int match(char *ss)
    {
        int len,now,res;
        len = strlen(ss),now = root,res = 0;
        for(int i = 0; i < len; i++)
        {
            int idx = getidx(ss[i]);
            int tmp = now = next[now][idx];
            while(tmp)
            {
                res += end[tmp];
                end[tmp] = 0;//°´ÌâÄ¿ÐÞ¸Ä
                tmp = fail[tmp];
            }
        }
        return res;
    }

    int go(int n,int m,int *v)
    {
        int ans=-0x3f3f3f3f,now=1,pre=0;
        memset(dp[pre],0,sizeof dp[pre]);
        dp[0][0][0]=1;
        for(int i=0,mx=1<<m;i<n;i++)
        {
            memset(dp[now],0,sizeof dp[now]);
            for(int j=0;j<tot;j++)
            for(int s=0;s<mx;s++)
            for(int k=0;dp[pre][j][s]&&k<LetterSize;k++)
                dp[now][next[j][k]][s|end[next[j][k]]]=1;
            swap(now,pre);
        }
        for(int i=0,mx=1<<m;i<tot;i++)
        for(int j=0;j<mx;j++)
        if(dp[pre][i][j])
        {
            int tmp=0;
            for(int k=0;k<m;k++)
            if(j&(1<<k))
                tmp+=v[k];
            ans=max(ans,tmp);
        }
        return ans;
    }
    void debug()
    {
        for(int i = 0;i < tot;i++)
        {
            printf("id = %3d,fail = %3d,end = %3d,chi = [",i,fail[i],end[i]);
            for(int j = 0;j < LetterSize;j++)
                printf("%3d",next[i][j]);
            printf("]
");
        }
    }
};
AC_auto ac;
char ss[2000];
int v[200];
int main(void)
{
    //freopen("in.acm","r",stdin);
    int n,m;
    while(~scanf("%d%d",&m,&n))
    {
        ac.init();
        for(int i=0;i<m;i++)
            scanf("%s%d",ss,v+i),ac.insert(ss,1<<i);
        ac.build();
        int ans=ac.go(n,m,v);
        if(ans<0) printf("No Rabbit after 2012!
");
        else printf("%d
",ans);
    }
    return 0;
}