hdu 4057 Rescue the Rabbit

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int root=0,N=4;
int n,m;
bool dp[2][1111][1111];
int c[1111][10],a[111];
int lon;
struct
{
int next[5],tmp,fail;
}trie[1111];

void trieini()
{
memset(trie,0,sizeof(trie));
lon=0;
}

void insert(char s[],int h)
{
int t=root;
int n=strlen(s+1);
for(int i=1;i<=n;i++)
{
if(trie[t].next[s[i]-'A']==0)
trie[t].next[s[i]-'A']=++lon;
t=trie[t].next[s[i]-'A'];
if(i==n)
trie[t].tmp=1<<h-1;
}
}

void getfail()
{
queue <int> q;
q.push(root);
while(!q.empty())
{
int t=q.front();
q.pop();
for(int i=0;i<N;i++)
if(trie[t].next[i])
{
int u=trie[t].next[i];
int tmp=trie[t].fail;
while(tmp!=root&&trie[tmp].next[i]==0)
tmp=trie[tmp].fail;
trie[u].fail=trie[tmp].next[i];
if(t==root) trie[u].fail=root;
trie[u].tmp|=trie[trie[u].fail].tmp;
q.push(u);
}
}
}

void work(char s[])
{
int n=strlen(s+1);
for(int i=1;i<=n;i++)
{
if(s[i]=='T')
s[i]='B';
else if(s[i]=='G')
s[i]='D';
}
}

void find()
{
for(int i=0;i<=lon;i++)
for(int p=0;p<4;p++)
{
int t=i;
while(t&&trie[t].next[p]==0)
t=trie[t].fail;
c[i][p]=trie[t].next[p];
}
}

int main()
{
while(scanf("%d %d",&n,&m)!=EOF)
{
trieini();
char s[111];
for(int i=1;i<=n;i++)
{
scanf("%s",s+1);
scanf("%d",&a[i]);
work(s);
insert(s,i);
}
getfail();
find();
memset(dp,0,sizeof(dp));
dp[0][0][0]=1;
for(int k=0;k<m;k++)
{
int now=k%2,to=(k+1)%2;
for(int i=0;i<(1<<n);i++)
for(int j=0;j<=lon;j++)
dp[to][i][j]=0;

for(int i=0;i<(1<<n);i++)
for(int j=0;j<=lon;j++)
if(dp[now][i][j])
for(int p=0;p<4;p++)
{
int u=c[j][p];
int v=trie[u].tmp;
dp[to][i|v][u]=1;
}
}
int to=m%2;
int ans=-1;
for(int i=0;i<(1<<n);i++)
for(int j=0;j<=lon;j++)
if(dp[to][i][j])
{
int sum=0;
for(int k=1;k<=n;k++)
{
int tmp=1<<k-1;
if(tmp&i)
sum+=a[k];
}
ans=max(ans,sum);
}
if(ans>=0)
printf("%d\n",ans);
else
printf("No Rabbit after 2012!\n");
}
return 0;
}