题目描述
有两个仅包含小写英文字母的字符串 A 和 B。
现在要从字符串 A 中取出 k 个互不重叠的非空子串,然后把这 k 个子串按照其在字符串A 中出现的顺序依次连接起来得到一个新的字符串。请问有多少种方案可以使得这个新串与字符串 BBB 相等?
注意:子串取出的位置不同也认为是不同的方案。
输入格式
第一行是三个正整数 n,m,k,分别表示字符串 A 的长度,字符串 B 的长度,以及问题描述中所提到的 k,每两个整数之间用一个空格隔开。
第二行包含一个长度为 n 的字符串,表示字符串 A。
第三行包含一个长度为 m 的字符串,表示字符串 B。
输出格式
一个整数,表示所求方案数。
由于答案可能很大,所以这里要求输出答案对 1000000007 取模的结果。
输入输出样例
输入 #1
6 3 1
aabaab
aab
输出 #1
2
输入 #2
6 3 2
aabaab
aab
输出 #2
7
输入 #3
6 3 3
aabaab
aab
输出 #3
7
说明/提示
对于所有 10 组数据:1≤n≤1000,1≤m≤200,1≤k≤m。
解释:简单dp,设dp[k][i][j]:表示串A的1-i,串B的1-j其中对串B分为k块,全都匹配的个数。则
d p [ k ] [ i ] [ j ] = d p [ k ] [ i − 1 ] [ j ] + d p [ k − 1 ] [ i − 1 ] [ j − 1 ] + . . d p [ k − 1 ] [ i ‘ ] [ j ‘ ] dp[k][i][j]=dp[k][i-1][j]+dp[k-1][i-1][j-1]+..dp[k-1][i^`][j^`] dp[k][i][j]=dp[k][i−1][j]+dp[k−1][i−1][j−1]+..dp[k−1][i‘][j‘],其中第二项开始加的要求是A[i]==B[j],A[i-1]==B[j-1],依次类推就好了,这里做了个优化,发现可以直接用前一个状态的表示那些求和,即dp[k][i-1][j-1]表示,那么最终方程就是
d p [ k ] [ i ] [ j ] = d p [ k ] [ i − 1 ] [ j ] + d p [ k − 1 ] [ i − 1 ] [ j − 1 ] + d p [ k ] [ i − 1 ] [ j − 1 ] − d p [ k ] [ i − 2 ] [ j − 1 ] dp[k][i][j]=dp[k][i-1][j]+dp[k-1][i-1][j-1]+dp[k][i-1][j-1]-dp[k][i-2][j-1] dp[k][i][j]=dp[k][i−1][j]+dp[k−1][i−1][j−1]+dp[k][i−1][j−1]−dp[k][i−2][j−1]
同时把第一维用滚动数组来维护,达到优化空间的目的。
#include<iostream>
#include<cstring>
#define N 1003
#define M 203
#define mod 1000000007
using namespace std;
int dp[3][N][M]={0};
char A[N],B[M];
int n=0,m=0,K=0;
int main(){
ios::sync_with_stdio(false);
cin>>n>>m>>K;
cin>>(A+1);cin>>(B+1);
for(int i=0;i<=n;i++) dp[0][i][0]=1;
for(int k=1;k<=K;k++){
memset(dp[k&1],0,sizeof(dp[k&1]));
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(j>i) break;
dp[k&1][i][j]+=dp[k&1][i-1][j];
dp[k&1][i][j]%=mod;
if(A[i]==B[j]){
dp[k&1][i][j]+=(dp[(k-1)&1][i-1][j-1]+dp[k&1][i-1][j-1])%mod;
dp[k&1][i][j]%=mod;
if(i>1) dp[k&1][i][j]-=dp[k&1][i-2][j-1];
dp[k&1][i][j]%=mod;
dp[k&1][i][j]+=mod;
dp[k&1][i][j]%=mod;
}
dp[k&1][i][j]%=mod;
}
}
}
cout<<dp[K&1][n][m]<<endl;
return 0;
}