HDU 3341 状态压缩DP+AC自动机

 题目大意：

调整基因的顺序，希望使得最后得到的基因包含有最多的匹配串基因，使得所能达到的智商最高

 

这里很明显要用状态压缩当前AC自动机上点使用了基因的情况所能达到的最优状态

我最开始对于状态的保存是，针对基因的个数转化为最小的二进制个数保存，但是浪费了很多二进制位，比如8 -> 1000，那么之后的1001,1010...1111都没用到

就MLE了

然后自己只能在压缩

比如数量为2 ， 3 ， 4 ， 5

那么jz[0] = 1 , jz[1] = 2+1 , jz[2] = 2*1+3*3+1 , jz[3] = 2*1+3*3+4*12+1

那么可以根据这种进制进行解码转码了

那么状态数就能通过解码转码轻松获得了

可能自己写的比较挫吧，跑c++跑不过，g++倒是完全没问题

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include <queue>
using namespace std;
#define LL long long
#define clr(x) memset(x , 0 , sizeof(x))
#define MAX_SIZE 502
#define CHAR_SIZE 4
const int MAX_STATE = 100000;
int n;
char str[50];

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

struct AC_Machine{
    int sz , ch[MAX_SIZE][CHAR_SIZE] , fail[MAX_SIZE], val[MAX_SIZE] , gene[MAX_SIZE];
    void init(){
        sz = 1;
        clr(ch[0]) , clr(val);
        gene[0] = 0;
    }

    void insert(char *s){
        int n=strlen(s) , u=0;
        for(int i=0 ; i<n ; i++){
            int c = Hash(s[i]);
            if(!ch[u][c]){
                clr(ch[sz]);
                val[sz] = gene[sz] = 0;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        gene[u] ++;
        val[u] = 1;
    }

    void get_fail(){
        queue<int> Q;
        fail[0] = 0;
        for(int c=0 ; c<CHAR_SIZE ; c++){
            int u = ch[0][c];
            if(u) {Q.push(u);fail[u]=0;}
        }
        while(!Q.empty()){
            int r = Q.front();
            val[r] |= val[fail[r]];
            Q.pop();
            for(int c=0 ; c<CHAR_SIZE ; c++){
                int u = ch[r][c];
                if(!u){ch[r][c]=ch[fail[r]][c] ; continue;}
                fail[u] = ch[fail[r]][c];
                Q.push(u);
            }
        }
    }
}ac;

int dp[MAX_SIZE][MAX_STATE];
int cnt[4] , jz[4];
int code[4] , save[4];
void decode(int state)
{
    for(int i=3 ; i>=0 ; i--){
        code[i] = state/jz[i];
        state -= code[i]*jz[i];
    }
}

int encode(int *code)
{
    int ret = 0;
    for(int i=3 ; i>=0 ; i--) ret+=code[i]*jz[i];
    return ret;
}

void DP()
{
    int all = encode(cnt);
    for(int i=0 ; i<=all ; i++)
        for(int j=0 ; j<ac.sz ; j++) dp[j][i] = -1;
    dp[0][0] = 0;
    for(int i=0 ; i<=all ; i++){
        for(int j=0 ; j<ac.sz ; j++){
            if(dp[j][i]<0) continue;
            decode(i);
            for(int k=0 ; k<4 ; k++){
                if(code[k]>=cnt[k]) continue;
                int v = ac.ch[j][k];
                int tmp = v , num=0;
                while(tmp&&ac.val[tmp]){
                    if(ac.gene[tmp]) num+=ac.gene[tmp];
                    tmp = ac.fail[tmp];
                }
                code[k]++;
                int newst = encode(code);
                code[k]--;
                dp[v][newst] = max(dp[j][i]+num , dp[v][newst]);
            }
        }
    }
}

void solve()
{
    int  len = strlen(str);
    memset(cnt , 0 , sizeof(cnt));
    for(int i=0 ; i<len ; i++) cnt[Hash(str[i])]++;
    jz[0] = 1 , jz[1] = cnt[0]+1 , jz[2] = 1+cnt[0]+cnt[1]*jz[1] , jz[3] = cnt[2]*jz[2]+jz[2];
    DP();
}

int main()
{
   // freopen("in.txt" , "r" , stdin);
    int cas = 0;
    while(scanf("%d" , &n) , n)
    {
        ac.init();
        for(int i=0 ; i<n ; i++){
            scanf("%s" , str);
            ac.insert(str);
        }
        ac.get_fail();
        scanf("%s" , str);
        solve();
        int all = encode(cnt);
        int ret = -1;
        for(int i=0 ; i<ac.sz ; i++) ret = max(ret , dp[i][all]);
        printf("Case %d: %d\n" , ++cas , ret);
    }
    return 0;
}