题意
有一些字符串有权值,生成长度为L的字符串,问最大权值。如果最大权值为负,则输出“No Rabbit after 2012!”。
题解
也没太多好说的,算是基础的AC自动机+DP吧。主要是每个字符串只计算一次权值比较坑。最开始的时候,我是把这些权值记录到一个结构体内,在DP的时候进行计算。但是这样会存在一些问题,比如说一个节点包含多个权值(包含多个字符串)的时候就很难进行计算。我最开始是判定已选择的字符串编号是否包含这个节点包含的所有字符串编号。这样的算法现在看来明显是错误的,我却在上面WA了好久。ORZ。。
正解是先不管权值的问题,先进行状态转移,看最后长度为L的时候能转移到哪几个状态。最后计算每个可以达到的状态的值,取最大值就可以了。
代码
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
#define MAXN 100010
#define MOD 1000000007
#define EPS 1e-10
using namespace std;
struct Node {
int id;
};
char st[120];
int ch[1210][4],f[1210];
Node val[1210];
int sz;
int dp[1310][1210];
int temp[1310][1210];
int num[15];
int n,l,nowVal,cnt;
int getNum(char c) {
if(c=='A') {
return 0;
} else if(c=='C') {
return 1;
} else if(c=='T') {
return 2;
} else if(c=='G') {
return 3;
}
}
void insert() {
int u=0;
int len=strlen(st);
UP(i,0,len) {
int x=getNum(st[i]);
if(!ch[u][x]) {
ch[u][x]=sz++;
}
u=ch[u][x];
}
val[u].id|=(1<<cnt);
}
void getFail() {
MEM(f,0);
queue<int> q;
UP(i,0,4) {
int x=ch[0][i];
if(x) {
q.push(x);
}
}
W(!q.empty()) {
int r=q.front();
q.pop();
UP(i,0,4) {
int x=ch[r][i];
if(!x) {
ch[r][i]=ch[f[r]][i];
continue;
}
q.push(x);
int v=f[r];
f[x]=ch[v][i];
val[x].id|=val[f[x]].id;
}
}
}
void solve() {
MEM(dp,0);
MEM(temp,0);
dp[0][0]=1;
UP(i,0,l) {
UP(s,0,1<<n) {
UP(j,0,sz) {
if(dp[j][s]==0)
continue;
UP(k,0,4) {
int x=ch[j][k];
temp[x][s|val[x].id]=dp[j][s];
}
}
}
memcpy(dp,temp,sizeof(temp));
}
int mx=-1;
UP(i,0,sz) {
UP(j,0,1<<n) {
// cout<<i<<" "<<j<<endl;
if(dp[i][j]==0)
continue;
int sum=0;
UP(x,0,n){
if(j&(1<<x))
sum+=num[x];
}
mx=max(mx,sum);
}
}
if(mx<=-1) {
puts("No Rabbit after 2012!");
} else {
printf("%d\n",mx);
}
}
int main() {
W(~scanf("%d%d",&n,&l)) {
MEM(ch,0);
MEM(val,0);
sz=1;
UP(i,0,n) {
scanf("%s %d",st,&nowVal);
cnt=i;
num[i]=nowVal;
insert();
}
getFail();
solve();
}
}
/*
1 5
A -1
2 5
GGG 4
TGGGC 6
2 5
AAAA 5
AAAG 6
*/