Luogu P2679 子串(字符串+dp)

P2679 子串

题意

题目描述

有两个仅包含小写英文字母的字符串\(A\)和\(B\)。

现在要从字符串\(A\)中取出\(k\)个互不重叠的非空子串，然后把这\(k\)个子串按照其在字符串\(A\)中出现的顺序依次连接起来得到一个新的字符串。请问有多少种方案可以使得这个新串与字符串\(B\)相等？

注意：子串取出的位置不同也认为是不同的方案。

输入输出格式

输入格式：

第一行是三个正整数\(n,m,k\)，分别表示字符串\(A\)的长度，字符串\(B\)的长度，以及问题描述中所提到的\(k\)，每两个整数之间用一个空格隔开。

第二行包含一个长度为\(n\)的字符串，表示字符串\(A\)。

第三行包含一个长度为\(m\)的字符串，表示字符串\(B\)。

输出格式：

一个整数，表示所求方案数。

由于答案可能很大，所以这里要求输出答案对\(1000000007\)取模的结果。

输入输出样例

输入样例：

6 3 1
aabaab
aab

输出样例：

2

输入样例：

6 3 2
aabaab
aab

输出样例：

7

输入样例：

6 3 3
aabaab
aab

输出样例：

7

说明

对于第\(1\)组数据:\(1 \leq n \leq 500,1 \leq m \leq 50,k=1\);

对于第\(2\)组至第\(3\)组数据:\(1 \leq n \leq 500,1 \leq m \leq 50,k=2\);

对于第\(4\)组至第\(5\)组数据:\(1 \leq n \leq 500,1 \leq m \leq 50,k=m\);

对于第\(1\)组至第\(7\)组数据:\(1 \leq n \leq 500,1 \leq m \leq 50,1 \leq k \leq m\);

对于第\(1\)组至第\(9\)组数据:\(1 \leq n \leq 1000,1 \leq m \leq 100,1 \leq k \leq m\);

对于所有\(10\)组数据:\(1 \leq n \leq 1000,1 \leq m \leq 200,1 \leq k \leq m\)。

思路

你可以看一篇优秀的博客。 --alecli

这位神犇叫为了我这道题。

设计状态\(dp[i][j][k][0/1]\)，\(i\)表示\(A\)字符串的前\(i\)位，\(j\)表示\(B\)字符串的前\(j\)位，\(k\)表示选取了多少个子串，\(0/1\)表示当前字符有没有选入子串中。

如果该位没有选，那么转移是显然易见的：

\[dp[i][j][k][0]=dp[i-1][j-1][k][1]+dp[i-1][j-1][k][0]
\]

它表示不论前一位选与不选，我都加一个空格，分开上一子串和下一子串

而如果要选这一位，就要分类讨论这一位上的\(A\)与\(B\)是否相同。

• 如果不同，那么\(dp[i][j][k][0]=0\)；
• 如果相同，那么\(dp[i][j][k][0]=dp[i-1][j-1][k][1]+dp[i-1][j-1][k-1][1]+dp[i-1][j-1][k-1][0]\)，它表示继续下一子串、在上一子串连续的情况下重新开始新一子串、直接作为新子串的开头。

那么答案就是\(dp[n][m][k][0]+dp[n][m][k][1]\)了。

顺便，我的代码怕空间不足，写了滚动数组。如果不写的话，要记得初始化\(dp\)数组的值。

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL P=1000000007;
LL n,m,k,dp[2][202][202][2];
string a,b;
int main()
{
    cin>>n>>m>>k;
    cin>>a>>b;
    a=' '+a;
    b=' '+b;
    dp[0][0][0][0]=dp[1][0][0][0]=1;
    for(LL i=1;i<=n;i++)
        for(LL j=1;j<=m;j++)
            for(LL p=1;p<=k;p++)
            {
                dp[i&1][j][p][0]=(dp[(i-1)&1][j][p][0]+dp[(i-1)&1][j][p][1])%P;
                if(a[i]==b[j]) dp[i&1][j][p][1]=(dp[(i-1)&1][j-1][p][1]+dp[(i-1)&1][j-1][p-1][0]+dp[(i-1)&1][j-1][p-1][1])%P;
                else dp[i&1][j][p][1]=0;
            }
    printf("%lld",(dp[n&1][m][k][1]+dp[n&1][m][k][0])%P);
    return 0;
}