洛谷P2679 子串

题目

因为该题需要求解方案数,因此我们不能用计数题的常用方法,即用递推来解决,类似于动态规划的思想。

我们用(dp[i][j][h][o])表示第一个串枚举到了(i),第二个串枚举到了(j)(A)中取了(h)串,(o)表示选不选当前第i位的方案数。

可以得出以下的状态转移:

如果此位不选择的话

(dp[i][j][h][0]=dp[i-1][j][h][0]+dp[i-1][j][h][1]),此位不选则用加法原理选之前一位选择的和。

如果此位选择的话:

如果(a[i])(b[j])不相等的话选择该位不能产生一个满足条件的子串。

如果相等的话,子串个数增加和不增加的情况都需要考虑,子串个数不增加的话,则(i-1)必须选,而选了(i)之后子串个数增加的话,(i-1)可选可不选。注意此时第二个串应该加上枚举到(j-1)的方案。

然后i的转移只跟i-1有关,因此考虑空间优化

#include <bits/stdc++.h>
#define N 1001001
#define mod 1000000007
using namespace std;
int n, m, k, dp[2][1001][1001][2];
char a[N], b[N];
void init()
{
	scanf("%d%d%d", &n, &m, &k);
	scanf("%s%s", a + 1, b + 1);
	dp[0][0][0][0] = dp[1][0][0][0] = 1;
}
int add(int a, int b) {return (a + b) % mod;}
int main()
{ 
 	init();
 	for (int i = 1, now = 1; i <= n; i++, now ^= 1)
 		for (int j = 1; j <= m; j++)
 			for (int h = 1; h <= k; h++)
 			{
 				dp[now][j][h][0] = (dp[now ^ 1][j][h][1] + dp[now ^ 1][j][h][0]) % mod;//上一位
 				if (a[i] == b[j])
 				{
 					dp[now][j][h][1] = 0;
 					dp[now][j][h][1] = (dp[now][j][h][1] + dp[now ^ 1][j - 1][h][1]) % mod;
 					dp[now][j][h][1] = (dp[now][j][h][1] + dp[now ^ 1][j - 1][h - 1][1]) % mod;
 					dp[now][j][h][1] = (dp[now][j][h][1] + dp[now ^ 1][j - 1][h - 1][0]) % mod;
 				}
 				else dp[now][j][h][1] = 0;
 			}
 	printf("%d", (dp[n & 1][m][k][0] + dp[n & 1][m][k][1])% mod);//第一位判断是否是奇数或偶数,最后一位判断是否相等。 
}
原文地址:https://www.cnblogs.com/liuwenyao/p/11829333.html