2021牛客暑期多校训练第5场--D--Double String

最新推荐文章于 2026-09-02 22:09:47 发布

原创最新推荐文章于 2026-09-02 22:09:47 发布 · 143 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

算法练习

本文介绍了一种基于字符串匹配的问题解决方法，通过动态规划求解两个字符串间的特定匹配组合数量。文章详细阐述了算法的设计思路，包括如何计算公共子序列个数及组合数，并提供了完整的代码实现。

题意

原题链接

给两个字符串A，B，从 {1 , 2 , 3 , … |A|} 里选出子序列 a ，从{ 1 , 2 , 3 , …|B| } 里选出子序列 b，求有多少个这种 a , b 组合满足条件。
条件为：

a与b长度相等
存在 i 使得 A $_{ai}$ < B $_{bi}$ , 且任意 j $\in$ (1, i )都有 A $_{aj}$ = B $_{bj}$ 成立。

题解

在 [1 , $i$ - 1] 和 [1, $j - 1$ ] 区间里找公共子序列的个数。
在 $i$ 这个点A[ $i$ ] < B[ $j$ ]。
$i$ 、 $j$ 后面的可以任意取。
假设 A 串后面还有 n 个任意取的长度，B串后还有 m 个任意取的长度。（n < m)
则任意取的个数为 $\sum_1^n C_n^iC_m^i$
根据组合数公式，上式等于 $C_{n+m}^m$
证明如下

注释

dp求公共子序列个数时dp [ $i$ ] [ $j$ ]表示的状态是在 $A$ 串的 [ ${1,i}$ ] 区间内和 $B$ 串的 [ $1, j$ ] 区间内公共子序列的个数。
求组合数时先将 $1$ ~ $N$ 内数的阶乘打表，存在 $f u n c$ 数组内。将在 $1$ ~ $N$ 内数的阶乘的逆元打表，存到 $i n v$ 数组内。
线性求逆元+求组合数参考数论知识

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5050;
const int P=1e9+7;
ll dp[N][N],func[N+N],inv[N+N];
ll power(ll a,ll b,ll P){
	ll res=1;
	while(b){
		if(b&1)
		{
			res*=a;
			res%=P;
		}
		a*=a;
		a%=P;
		b>>=1;
	}
	return res;
}
void init(){
	func[0]=1;
	for(ll i=1;i<N+N;i++){
		func[i]=func[i-1]*i%P;
	}
	inv[1]=1;
	inv[0]=1;
	for(ll i=2;i<N+N;i++){
		inv[i]=(P-P/i)*inv[P%i]%P;
	}
	for(int i=2;i<2*N;i++){
        inv[i]=(inv[i-1]*inv[i])%P;
    }
}
ll C(ll n,ll m,ll P){
	if(m>n)
	return 1;
	ll res=1;
	res=func[n]*inv[m]%P*inv[n-m]%P; 
	return res;
 } 
int main()
{
	init();
	char a[N],b[N];
	cin>>a+1>>b+1;
	ll l1=strlen(a+1);
	ll l2=strlen(b+1);
	for(ll i=1;i<=l1;i++){
		for(ll j=1;j<=l2;j++){
			dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1];
			if(a[i]==b[j])
			{
				dp[i][j]+=dp[i-1][j-1]+1;
			}
			dp[i][j]=(dp[i][j]+P)%P;
		}
	}
	ll res=0;
	for(int i=1;i<=l1;i++)
	for(int j=1;j<=l2;j++){
		if(a[i]<b[j]){
			ll n=l1-i;
			ll m=l2-j;
			if(m>n)
			swap(n,m);
			res+=(dp[i-1][j-1]+1)*C(n+m,m,P)%P;
			res%=P; 
		}
	}
	cout<<res<<endl;
	
	return 0;
}

标签

#算法 #动态规划