HDU 4057 Rescue the Rabbit（自动机+DP）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4057

题意：给定m个串，每个串有一个价值。构造一个长度为n的串，包含了m个串中的某个串则加上相应的价值。使得总价值最大？

思路：f[i][j][k]表示长度为i、包含m个串的状态为j、在自动机上的位置为k。

struct node
{
    int next[4],fail,val,st;

    void init()
    {
        clr(next,0);
        fail=0;
        val=st=0;
    }
};

node a[N];
int e,n,m;
int mp[130];

void init()
{
    mp['A']=0;
    mp['T']=1;
    mp['G']=2;
    mp['C']=3;
}

void insert(char s[],int st)
{
    int i,k,p=0;
    for(i=0;s[i];i++)
    {
        k=mp[s[i]];
        if(a[p].next[k]==0)
        {
            a[e].init();
            a[p].next[k]=e++;
        }
        p=a[p].next[k];
    }
    a[p].st|=1<<st;
}

queue<int> Q;

void build()
{
    int i,j,k,p,q;
    FOR0(i,4) if(a[0].next[i]) Q.push(a[0].next[i]);
    while(!Q.empty())
    {
        k=Q.front();
        Q.pop();
        for(i=0;i<4;i++)
        {
            if(a[k].next[i])
            {
                p=a[k].next[i];
                q=a[k].fail;
                Q.push(p);
                a[p].fail=a[q].next[i];
                a[p].st|=a[a[p].fail].st;
            }
            else
            {
                q=a[k].fail;
                a[k].next[i]=a[q].next[i];
            }
        }
    }
}

char s[50005];
int f[2][1030][1005],val[N];


int get(int st)
{
    int ans=0,i;
    FOR0(i,m) if(st&(1<<i)) ans+=val[i];
    return ans;
}

void DP()
{
    int pre=0,cur=1,i,j,k,t,p,st,val,x=(1<<m);
    clr(f[pre],0);
    f[0][0][0]=1;
    FOR0(i,n)
    {
        clr(f[cur],0);
        FOR0(j,x) FOR0(k,e) if(f[pre][j][k]) FOR0(t,4)
        {
            p=a[k].next[t];
            st=j|a[p].st;
            f[cur][st][p]=1;
        }
        pre^=1;
        cur^=1;
    }
    int ans=-1;
    FOR0(i,x) FOR0(j,e) if(f[pre][i][j])
    {
        ans=max(ans,get(i));
    }
    if(ans>=0) PR(ans);
    else puts("No Rabbit after 2012!");
}

int main()
{
    init();
    while(scanf("%d%d",&m,&n)!=-1)
    {
        a[0].init();e=1;
        int i,x;
        FOR0(i,m) RD(s),RD(val[i]),insert(s,i);
        build();
        DP();
    }
    return 0;
}