Luogu 2679 子串 (动态规划)
Luogu 2679 NOIP 2015 子串 (动态规划)
Description
有两个仅包含小写英文字母的字符串 A 和 B。现在要从字符串 A 中取出 k 个互不重叠的非空子串,然后把这 k 个子串按照其在字符串 A 中出现的顺序依次连接起来得到一 个新的字符串,请问有多少种方案可以使得这个新串与字符串 B 相等?注意:子串取出 的位置不同也认为是不同的方案。
Input
第一行是三个正整数 n,m,k,分别表示字符串 A 的长度,字符串 B 的长度,以及问
题描述中所提到的 k,每两个整数之间用一个空格隔开。 第二行包含一个长度为 n 的字符串,表示字符串 A。 第三行包含一个长度为 m 的字符串,表示字符串 B。
Output
输出共一行,包含一个整数,表示所求方案数。由于答案可能很大,所以这里要求输出答案对 1,000,000,007 取模的结果。
Sample Input
6 3 1
aabaab
aab
Sample Output
2
Http
Luogu:https://www.luogu.org/problem/show?pid=2679
Source
动态规划
解决思路
设\(F[i][j][k][0/1]\)表示当前选到\(A\)的第\(i\)位,匹配到\(B\)的第\(j\)位,是第\(k\)个子串。0表示\(A\)的该位必选,1表示选or不选都行。
考虑两种情况
若\(A[i]==B[j]\),则可以将新的这个字符串接在前一个子串后面,这时要求前面一个必选,则\(F[i][j][k][0]\)可以从\(F[i-1][j-1][k][0]\)转移过来,也可以新开一个子串,此时对前面的字符是否选取没有要求,所以可以从\(F[i-1][j-1][k-1][1]\)转移来。
若\(A[i]!=B[j]\)则\(F[i][j][k][0]=0\)。
在求完\(F[i][j][k][0]\)后,\(F[i][j][k][1]=F[i][j][k][0]+F[i-1][j][k][1]\)。注意滚动数组。
代码
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; const int maxN=1010; const int maxM=210; const int Mod=1000000007; const int inf=2147483647; int n,m,K; char A[maxN]; char B[maxM]; int F[2][maxM][maxM][2]; int main() { scanf("%d%d%d",&n,&m,&K); scanf("%s",(A+1)); scanf("%s",(B+1)); memset(F,0,sizeof(F)); F[0][0][0][1]=F[1][0][0][1]=1;//初始值 for (int i=1;i<=n;i++) { F[i%2][0][0][1]=1; for (int j=1;j<=m;j++) for (int k=1;k<=K;k++) { if (A[i]==B[j]) F[i%2][j][k][0]=(F[(i-1)%2][j-1][k][0]+F[(i-1)%2][j-1][k-1][1])%Mod; else F[i%2][j][k][0]=0; F[i%2][j][k][1]=(F[i%2][j][k][0]+F[(i-1)%2][j][k][1])%Mod; } } printf("%d\n",F[n%2][m][K][1]%Mod); return 0; }