AC自动机+状压DP,这个相当明显
状态设置为dp[len][u][st]表示在自动机内走了len步后,位于结点u,经过的pattern的状态为st时,所能取得的最大value。
然后就很容易得出状态方程了。
这里需要用到滚动数组,否则MLE。
刚开始用的是递归的方式写的,坑了2炮,最后没办法,只能改成迭代写。
因为平时迭代写的不多,所以改得比较慢,调了好久才过,这里以后需要注意!
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int sigma=4;
const int maxn=1010;
const int inf=1<<30;
int ch[maxn][sigma],val[maxn];
int sz;
void init(){sz=1;memset(ch[0],0,sizeof(ch[0]));}
int idx(char c)
{
if(c=='A') return 0;
else if(c=='T') return 1;
else if(c=='G') return 2;
else return 3;
}
void insert(char *s,int v)
{
int u=0,n=strlen(s);
for(int i=0;i<n;i++)
{
int c=idx(s[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
val[u]|=(1<<v);
}
int f[maxn];
void getfail()
{
queue<int> q;
f[0]=0;
for(int c=0;c<sigma;c++)
{
int u=ch[0][c];
if(u)
{
f[u]=0;q.push(u);
}
}
while(!q.empty())
{
int r=q.front();q.pop();
for(int c=0;c<sigma;c++)
{
int u=ch[r][c];
if(!u)
{
ch[r][c]=ch[f[r]][c];
continue;
}
q.push(u);
int v=f[r];
while(v && !ch[v][c]) v=f[v];
f[u]=ch[v][c];
val[u]|=val[f[u]];
}
}
}
int n,l;
int w[20],v[1<<12];
char str[maxn];
int dp[2][1010][1<<10];
void getval()
{
for(int i=0;i<(1<<n);i++)
{
v[i]=0;
for(int j=0;j<n;j++) if(i&(1<<j)) v[i]+=w[j];
}
}
int main()
{
//freopen("data","r",stdin);
while(~scanf("%d%d",&n,&l))
{
memset(val,0,sizeof(val));
init();
for(int i=0;i<n;i++)
{
scanf("%s%d",str,&w[i]);
insert(str,i);
}
getval();
getfail();
//for(int i=0;i<sz;i++) cout<<i<<" "<<val[i]<<endl;
int flag=0;
for(int i=0;i<sz;i++)
for(int st=0;st<(1<<n);st++) dp[flag][i][st]=-inf;
dp[0][0][0]=0;
for(int i=0;i<=l;i++)
{
for(int j=0;j<sz;j++)
for(int st=0;st<(1<<n);st++) dp[flag^1][j][st]=-inf;
for(int j=0;j<sz;j++)
{
for(int st=0;st<(1<<n);st++)
{
for(int k=0;k<4;k++)
{
int &tmp=dp[flag^1][ch[j][k]][st|val[j]];
tmp=max(tmp,dp[flag][j][st]+v[val[j]&(~st)]);
}
}
}
flag^=1;
}
int ans=-inf;
for(int i=0;i<sz;i++)
for(int st=0;st<(1<<n);st++)
{
ans=max(ans,dp[flag][i][st]);
}
if(ans<0) puts("No Rabbit after 2012!");
else printf("%d\n",ans);
}
return 0;
}