牛客2021暑期训练5-D-Double Strings

牛客2021暑期训练5-D-Double Strings

题目链接

题意

对所给的两个字符串 A、B 求好的子序列个数，好的子序列为：一段相同的前缀（可以无）+ 一个不同字符（a比b小）+ 长度相同的任意后缀（可以无）

题解

设 $f[i][j]$ 为 $A[1...i]$ 和 $B[1...j]$ 可用前缀个数，可得递推式
$f[i][j]=f[i-1][j]+f[i][j-1]-f[i-1][j-1]$
当 A[i]==B[i] 时，额外有 $f[i][j]+=f[i-1][j-1]+1$

设 $g[i][j]$ 为从 {1, 2, …, i} 和 {1, 2, …, j} 中取出长度相同的序列的个数，等价于所有字符相等的 $f$，即
$g[i][j]=g[i-1][j]+g[i][j-1]+1$

因此可以先处理 $f$ 、 $g$ 数组，再枚举 $i，j$，当 $A[i]<B[j]$ 时，$ans+=(f[i-1][j-1]+1)\times (g[la-i][lb-j]+1)$ ，加1是“可以无”

代码

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int N=5e3+9,mod=1e9+7;
int la,lb;
LL ans;
int f[N][N],g[N][N];
char a[N],b[N];
int main()
{
	cin>>a+1>>b+1;
	la=strlen(a+1);
	lb=strlen(b+1);
	for(int i=1;i<=la;i++)
		for(int j=1;j<=lb;j++)
		{
			f[i][j]=(f[i-1][j]+f[i][j-1]-f[i-1][j-1])%mod;
			if(a[i]==b[j]) f[i][j]=(f[i][j]+f[i-1][j-1]+1)%mod;
			g[i][j]=(g[i-1][j]+g[i][j-1]+1)%mod;
		}
	for(int i=1;i<=la;i++)
		for(int j=1;j<=lb;j++)
			if(a[i]<b[j])
				ans=(ans+1LL*(f[i-1][j-1]+1)*(g[la-i][lb-j]+1)%mod)%mod;
	printf("%lld",(ans+mod)%mod);
	return 0;
}