hdu 4057 Rescue the Rabbit

ac自动机的状压dp

dp[i][j][k]表示长度为i，包含字符串的状态为j，自动机的状态为k是否可行。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int root=0,N=4;
int n,m;
bool dp[2][1111][1111];
int c[1111][10],a[111];
int lon;
struct
{
    int next[5],tmp,fail;
}trie[1111];

void trieini()
{
    memset(trie,0,sizeof(trie));
    lon=0;
}

void insert(char s[],int h)
{
    int t=root;
    int n=strlen(s+1);
    for(int i=1;i<=n;i++)
    {
        if(trie[t].next[s[i]-'A']==0)
        trie[t].next[s[i]-'A']=++lon;
        t=trie[t].next[s[i]-'A'];
        if(i==n)
        trie[t].tmp=1<<h-1;
    }
}

void getfail()
{
    queue <int> q;
    q.push(root);
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        for(int i=0;i<N;i++)
        if(trie[t].next[i])
        {
            int u=trie[t].next[i];
            int tmp=trie[t].fail;
            while(tmp!=root&&trie[tmp].next[i]==0)
            tmp=trie[tmp].fail;
            trie[u].fail=trie[tmp].next[i];
            if(t==root) trie[u].fail=root;
            trie[u].tmp|=trie[trie[u].fail].tmp;
            q.push(u);
        }
    }
}

void work(char s[])
{
    int n=strlen(s+1);
    for(int i=1;i<=n;i++)
    {
        if(s[i]=='T')
        s[i]='B';
        else if(s[i]=='G')
        s[i]='D';
    }
}

void find()
{
    for(int i=0;i<=lon;i++)
    for(int p=0;p<4;p++)
    {
        int t=i;
        while(t&&trie[t].next[p]==0)
        t=trie[t].fail;
        c[i][p]=trie[t].next[p];
    }
}

int main()
{
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        trieini();
        char s[111];
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s+1);
            scanf("%d",&a[i]);
            work(s);
            insert(s,i);
        }
        getfail();
        find();
        memset(dp,0,sizeof(dp));
        dp[0][0][0]=1;
        for(int k=0;k<m;k++)
        {
            int now=k%2,to=(k+1)%2;
            for(int i=0;i<(1<<n);i++)
            for(int j=0;j<=lon;j++)
            dp[to][i][j]=0;

            for(int i=0;i<(1<<n);i++)
            for(int j=0;j<=lon;j++)
            if(dp[now][i][j])
            for(int p=0;p<4;p++)
            {
                int u=c[j][p];
                int v=trie[u].tmp;
                dp[to][i|v][u]=1;
            }
        }
        int to=m%2;
        int ans=-1;
        for(int i=0;i<(1<<n);i++)
        for(int j=0;j<=lon;j++)
        if(dp[to][i][j])
        {
            int sum=0;
            for(int k=1;k<=n;k++)
            {
                int tmp=1<<k-1;
                if(tmp&i)
                sum+=a[k];
            }
            ans=max(ans,sum);
        }
        if(ans>=0)
        printf("%d\n",ans);
        else
        printf("No Rabbit after 2012!\n");
    }
    return 0;
}