HDU4057 Rescue the Rabbit（AC自动机+状压dp）

题意：

给你n个基因串和每个基因串拥有的价值，可以为正也可以为负，询问构造长度为m的基因价值最大为多少？每种基因如有多个，当作一个。

题解：

这种题很容易想到在AC自动机上DP，我们不妨设$dp[i][j][S]$表示当前处理到第$i$个字符，并且在AC自动机上的$j$点，已经构成的基因串集合为$S$，然后从父亲向下递推看能否到达下一个状态就好了，不过dp数组的空间$100\ast 1000\ast 2^{10}$确实超过了这道题的内存大小，只好滚动数组优化一下空间了。

---

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1005;
const int INF = 0x3f3f3f3f;
int nxt[MAXN][4],fail[MAXN],val[MAXN],w[MAXN];
bool dp[2][MAXN][1<<10];
int tot=0,n,m;
inline void Init(){
    for(int i=0;i<=tot;i++){
        fail[i]=val[i]=0;
        for(int j=0;j<4;j++)
            nxt[i][j]=0;
    }
    tot=0;
}
inline int idx(char ch){
    if(ch=='A') return 0; else if(ch=='C') return 1;
    else if(ch=='G') return 2; else return 3;
}
inline void Insert(char *s,int k){
    int root=0,len=strlen(s);
    for(int i=0;i<len;i++){
        if(!nxt[root][idx(s[i])]) nxt[root][idx(s[i])]=++tot;
        root=nxt[root][idx(s[i])];
    }
    val[root] |= (1<<k);
}
inline void Build(){
    queue<int> que;
    for(int i=0;i<4;i++) if(nxt[0][i]) que.push(nxt[0][i]);
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=0;i<4;i++){
            if(nxt[u][i]){
                fail[nxt[u][i]] = nxt[fail[u]][i];
                val[nxt[u][i]] |= val[fail[nxt[u][i]]];
                que.push(nxt[u][i]);
            }else{
                nxt[u][i] = nxt[fail[u]][i];
            }
        }
    }
}
inline int sum(int x){
    int ans=0;
    for(int i=0;i<=10;i++)
        if((x>>i)&1) ans+=w[i];
    return ans;
}
inline void solve(){
    memset(dp,false,sizeof(dp));
    dp[0][0][0]=true;
    for(int i=1;i<=m;i++){
        memset(dp[i&1],0,sizeof(dp[i&1]));
        for(int j=0;j<=tot;j++){
            for(int k=0;k<4;k++){
                for(int p=0;p<(1<<n);p++){
                    if(dp[(i+1)&1][j][p])
                        dp[i&1][nxt[j][k]][p|val[nxt[j][k]]]=true;
                }
            }
        }
    }
    int res=-INF;
    for(int j=0;j<(1<<n);j++){
        for(int i=0;i<=tot;i++){
            if(dp[m&1][i][j]){
                res=max(res,sum(j));
                break;
            }
        }
    }
    if(res<0) printf("No Rabbit after 2012!\n");
    else printf("%d\n",res);
}
char s[MAXN];
int main(){
    while(~scanf("%d%d",&n,&m)){
        Init();
        for(int i=0;i<n;i++) scanf("%s%d",s,&w[i]),Insert(s,i);
        Build();
        solve();
    }
    return 0;
}