BZOJ5337 || 洛谷P4591[TJOI2018]str【KMP+DP】

Time Limit: 10 Sec
Memory Limit: 256 MB

Description

小豆参加了生物实验室。在实验室里，他主要研究蛋臼质。他现在研究的蛋臼质是由k个氨基酸按一定顺序构成的。每一个氨基酸都可能有a种碱基序 列si_j 构成。现在小豆有一个碱基串s，小豆想知道在这个碱基上都多少中不同的组合方式可能得到这个蛋白质。即求由k段字符串有序合并成的字符串s1,有多少种不同方式能够匹配字符串s，其中k段字符串的选法不同，或者与s匹配上 的位置不同认为是不同的方式。

Input

第一行一个数，表示这个蛋臼质由k个氨基酸。
第二行一个字符串s，表示小豆现在有的碱基串。
第三行开始接下来k行表示第i个氨基酸可能的碱基序列，对于第i个氨基酸，ai表示这个氨基酸可能的碱基序列种数，
接下来ai个字符串表示这ai 种可能的碱基序列，用空格隔开。
1 ≤ k ≤ 100, |s| ≤ 10000,ai ≤ 10

Output

输出一个数目标是不同的方案数
(不同的方案数是指不同的子碱基串或者相同的碱基串不同的氨基酸排列方式)

---

题目分析

一开始搞错题意瞎调了好久
注意k个氨基酸连成的字符串只要是蛋白质的子串就行，不是要全部匹配

$dp[i][j]$表示考虑前$i$个氨基酸，当前匹配到蛋白质的第$j$为的不同方案数
每读入一个可能氨基酸串，就把它与蛋白质串进行KMP匹配
发现基酸串在蛋白质串中的一个匹配位置k就转移
$dp[i][k]+=dp[i-1][k-len]$
初始化$dp[0][j]=1$

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
#include<map>
using namespace std;
typedef long long lt;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const lt mod=1e9+7;
const int maxn=10010;
int nxt[maxn],m;
lt dp[110][maxn];
char txt[maxn],pt[maxn];

void qnxt()
{
	int n=strlen(pt+1);
	int j=0; nxt[1]=0;
	for(int i=2;i<=n;++i)
	{
		while(j!=0&&pt[i]!=pt[j+1]) j=nxt[j];
		if(pt[i]==pt[j+1]) ++j;
		nxt[i]=j;
	}
}

void KMP(int id)
{
	int n=strlen(pt+1);
	qnxt(); int j=0;
	for(int i=1;i<=m;++i)
	{
		while(j!=0&&pt[j+1]!=txt[i]) j=nxt[j];
		if(pt[j+1]==txt[i]) ++j;
		if(j==n) dp[id][i]=(dp[id][i]+dp[id-1][i-n])%mod;
	}
}

int main()
{
    int k=read(); 
    scanf("%s",txt+1); m=strlen(txt+1);
    for(int i=0;i<=m;++i) dp[0][i]=1;
    for(int i=1;i<=k;++i)
    {
    	int ai=read();
    	while(ai--){
    		scanf("%s",pt+1);
    		KMP(i);
		}
	}
	
	lt ans=0;
	for(int i=1;i<=m;++i)
	ans=(ans+dp[k][i])%mod;
	printf("%lld",ans);
    return 0;
}