Hduoj 2296 Ring

Ring

AC自动机+DP。方程很容易写出来，但是对于串的处理确实比较麻烦的。首先方程：dp[i][j] = max{ dp[i-1][k] + tag[j] },这里tag[j]表示到达j状态时所有的匹配得到的数目之和。在递推的时候我们需要注意如果一个状态没有被经过(或者说是没有到达)，那么是不能用这个状态来递推到下一个状态的。所以在写的时候有长度为i的到长度为i+1的会更容易写。然后就是字符串的比较问题，这里我直接使用string类，因而时间会相对慢很多。在处理的时候，如果使用滚动数组处理，那么对于dp[j]中的肯定是保存的最短的，且字典序最小的，但是如果采用普通的二维数组的话，那么存储的就是长度为i的了。我用二维数组写的，然后在找最后答案的时候，我们需要对整个二维数组进行一次扫描。这样才能找到答案。

/*
author    : csuchenan
PROG      : hduoj 2296
LANG      : C++
Algorithm : AC + DP
dp[i][j] = max{dp[i - 1][k] + tag[ next[k][c]]}
注意递推的时候，需要有前面的状态才能递推到后面的状态
*/
#include <cstdio>
#include <string>
#include <cstring>
#include <queue>
using namespace std ;

#define maxn 1101


int N ;
int L ;
int cnt ;

int next[maxn][26] ;
int fail[maxn] ;
int tag[maxn] ;
char alpa[101][11] ;

struct Node{
    int val  ;
    string s ;
}dp[55][maxn];

inline void initNode(Node &a){
    a.val = 0  ;
    a.s   = "" ;
}

inline void init(){
    memset(next , 0 , sizeof(next)) ;
    memset(fail , 0 , sizeof(fail)) ;
    memset(tag  , 0 , sizeof(tag))  ;
    cnt = 0 ;
}

void insert(char * str , int val){
    char * p = str ;
    //printf("*%s*\n" ,str) ;
    int b ;
    int c(0) ;

    while(*p){
        b = *p - 'a' ;
        if(!next[c][b])
            next[c][b] = ++ cnt ;
        c = next[c][b]  ;
        p ++ ;
    }
    tag[c] = val ;
}

void build_ac(){
    queue<int> Q ;
    int cur(0) ;
    int child  ;

    fail[cur] = 0 ;
    Q.push(cur) ;

    while( !Q.empty() ){
        cur = Q.front() ;
        Q.pop() ;
        for(int i = 0 ; i < 26 ; i ++){
            child = next[cur][i] ;
            if(child){
                Q.push(child) ;
                if(!cur)
                    fail[child] = 0 ;
                else{
                    int tmp = fail[cur] ;
                    for( ; tmp && !next[tmp][i] ; tmp = fail[tmp])
                        ;
                    if(next[tmp][i])
                        fail[child] = next[tmp][i] ;
                    else
                        fail[child] = 0 ;
                    tag[child] += tag[ fail[child] ] ;
                }
            }
            else{
                next[cur][i] = next[ fail[cur] ][i] ;
            }
        }
    }
}
void debug(){
    for(int i = 0 ; i <= L ; i ++){
        printf("%d :" , i ) ;
        for(int j = 0 ; j <= cnt ; j ++){
            printf("%d %s  " , dp[i][j].val , dp[i][j].s.c_str() ) ;
        }
        printf("\n") ;
    }
}
bool cmp(string a , string b){
    if(a.size() != b.size())
        return a.size() < b.size() ;
    return a < b ;
}
void solve(){
    for(int i = 0 ; i <= L ; i ++){
        for(int j = 0 ; j <= cnt ; j ++){
            initNode(dp[i][j]) ;
        }
    }

    for(int i = 0 ; i <= L ; i ++){
        for(int j = 0 ; j <= cnt ; j ++){
            //起始态只有0合法，其它无字符串的都是非法态
            if(j && dp[i][j].s == "")
                continue ;

            for(int k = 0 ; k < 26 ; k ++){

                int cur = next[j][k] ;
                int w = dp[i][j].val + tag[cur] ;

                if(w > dp[i + 1][cur].val){
                    string str = dp[i][j].s + (char)(k + 'a') ;
                    dp[i + 1][cur].val = w ;
                    dp[i + 1][cur].s   = str ;
                }
                else if(w == dp[i + 1][cur].val){
                    string str = dp[i][j].s + (char)(k + 'a') ;
                    if(dp[i + 1][cur].s == "" ||
                       (dp[i + 1][cur].s != "" &&
                        cmp(str , dp[i + 1][cur].s))
                    )
                        dp[i + 1][cur].s = str ;
                }
            }
        }
    }
    //debug() ;
    int ans(0);
    string anstr("") ;
    for(int j = 1 ; j <= L ; j ++){
        for(int i = 0 ; i <= cnt ; i ++){
            if(ans < dp[j][i].val){
                ans = dp[j][i].val ;
                anstr = dp[j][i].s ;
            }
            else if(ans == dp[j][i].val && cmp(dp[j][i].s , anstr)){
                anstr = dp[j][i].s ;
            }
        }
    }
    printf("%s\n" ,anstr.c_str()) ;
}

int main(){
    int T ;
    scanf("%d" , &T) ;
    while(T--){
        scanf("%d%d" , &L , &N) ;
        init() ;
        for(int i = 0 ; i < N ; i ++){
            scanf("%s" ,alpa[i]) ;
        }
        int d ;
        for(int i = 0 ; i < N ; i ++){
            scanf("%d" , &d) ;
            insert(alpa[i] , d) ;
        }
        build_ac() ;
        solve() ;
    }
    return 0 ;
}