洛谷-2679 子串

题目描述
有两个仅包含小写英文字母的字符串 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块，全都匹配的个数。则
$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]表示，那么最终方程就是
$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;
}