hdu4057(AC自动机+状态压缩dp)

#include <queue>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
int q[1005],dp[2][1005][(1<<10)+5];
int fail[100005],vis[100005],val[100005];
int str[100005][30],dis[100005];
int root,id;
int change(char c){
if(c=='A')return 0;
if(c=='T')return 1;
if(c=='G')return 2;
if(c=='C')return 3;
return 0;
}                                               //只有四个字母
void insert(char *s){
int u=0;
for(;*s;s++){
if(!str[u][change(*s)])
str[u][change(*s)]=root++;
u=str[u][change(*s)];
}
dis[u]|=(1<<(id-1));
id++;
}
void getfail(){
int u,v,i,l,r,temp;
l=0,r=0;
q[r++]=0;
while(l<r){
u=q[l++];
for(i=0;i<4;i++){
if(!str[u][i])
str[u][i]=str[fail[u]][i];
else
{
temp=str[u][i];
if(u)
fail[temp]=str[fail[u]][i];
dis[temp]|=dis[fail[temp]];
q[r++]=temp;
}
}
}
}                                               //自动机模板
int cal(int t,int m){
int i,ans;
ans=0;
for(i=0;i<m;i++)
if(t&(1<<i))
ans+=val[i];
return ans;
}                                               //计算出每个状态的价值
int main(){
int n,m,k,i,j,x,y,z,ans;
char s[1005];
while(scanf("%d%d",&m,&k)!=EOF){
root=id=1;
memset(str,0,sizeof(str));
memset(val,0,sizeof(val));
memset(dis,0,sizeof(dis));
memset(fail,0,sizeof(fail));
for(i=0;i<m;i++){
scanf("%s%d",s,&val[i]);
insert(s);
}
getfail();
memset(dp,0,sizeof(dp));
dp[0][0][0]=1;
for(x=0;x<k;x++){                           //空间不够，因此用滚动数组
memset(dp[(x+1)&1],0,sizeof(dp[(x+1)&1]));
for(y=0;y<root;y++)
for(z=0;z<(1<<m);z++)
if(dp[x&1][y][z])
for(j=0;j<4;j++){
dp[(x+1)&1][str[y][j]][z|dis[str[y][j]]]=dp[x&1][y][z];
}
}                                           //自动机+状态压缩dp的题还是很相似的
ans=-INF;
for(i=0;i<(1<<m);i++)
for(j=0;j<root;j++)
if(dp[k&1][j][i])
ans=max(ans,cal(i,m));
if(ans<0)
puts("No Rabbit after 2012!");
else
printf("%d\n",ans);
}
return 0;
}