hdu 4057 Rescue the Rabbit(AC自动机+状压dp)

题目链接：hdu 4057 Rescue the Rabbit

题意：

给出一些模式串，每个串有一定的价值，现在构造一个长度为M的串，问最大的价值为多少，每个模式串最多统计一次。

题解：

由于每个模式串最多统计一次，所以我们要考虑记录是否已经存在该串。

考虑dp[i][j][k]表示当前考虑到i的长度，存在的串的组合为j，在AC自动机上走到了k这个节点的状态。

然后转移一下就能将所有能到达的状态走到。然后取一个最大值就行了。

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int AC_N=20000,tyn=4;
struct AC_automation{
    int tr[AC_N][tyn],cnt[AC_N],Q[AC_N],fail[AC_N],tot;
    inline int getid(char x)
    {
        if(x=='A')return 0;
        if(x=='T')return 1;
        if(x=='G')return 2;
        return 3;
    }
    void nw(){cnt[++tot]=0,fail[tot]=0;memset(tr[tot],0,sizeof(tr[tot]));}
    void init(){tot=-1,fail[0]=-1,nw();}
    void insert(char *s,int idx,int x=0){
        for(int len=strlen(s),i=0,w;i<len;x=tr[x][w],i++)
            if(!tr[x][w=getid(s[i])])nw(),tr[x][w]=tot;
        cnt[x]|=1<<(idx-1);
    }
    void build(int head=1,int tail=0){
        for(int i=0;i<tyn;i++)if(tr[0][i])Q[++tail]=tr[0][i];
        while(head<=tail)for(int x=Q[head++],i=0;i<tyn;i++)
            if(tr[x][i])
            {
                fail[tr[x][i]]=tr[fail[x]][i],Q[++tail]=tr[x][i];
                cnt[tr[x][i]]|=cnt[fail[tr[x][i]]];
            }
            else tr[x][i]=tr[fail[x]][i];
    }
}AC;

int dp[2][1<<10][1001],n,m,val[20],now,U,ans;
char s[1000];

int getans(int u)
{
    int ans=0;
    for(int i=1;u;u>>=1,i++)if(u&1)ans+=val[i];
    return ans;
}

int main(){
    while(~scanf("%d%d",&n,&m))
    {
        now=0,ans=-INT_MAX;AC.init();
        F(i,1,n)
        {
            scanf("%s%d",s,val+i);
            AC.insert(s,i);
        }
        AC.build(),U=(1<<n)-1;
        mst(dp[now],0),dp[now][0][0]=1;
        F(i,1,m)
        {
            now^=1,mst(dp[now],0);
            F(r,0,U)F(j,0,AC.tot)
            if(dp[now^1][r][j])
                F(k,0,3)dp[now][r|AC.cnt[AC.tr[j][k]]][AC.tr[j][k]]=1;
        }
        F(j,0,U)F(i,0,AC.tot)if(dp[now][j][i]){ans=max(ans,getans(j));break;}
        if(ans>=0)printf("%d
",ans);
        else puts("No Rabbit after 2012!");
    }
    return 0;
}