链接:https://ac.nowcoder.com/acm/contest/11256/D
题解:
好的方案的构成是“一段相同的前缀+一个不同字符(a比b小)+长度相同的任意后缀”。枚举不同的字符在两个序列中的位置。
用dp[i][j]表示只考虑 A 中的前 i 个字符和 B 中的前 j 个字符时的相同的子序列的个数,转移可以 O(1),这样可以统计出相同的前缀个数,这部分是 O(|s|*|t|) 的。
长度相同的任意后缀也可以用类似的 dp 计算,或者设 A 中此时剩余长度为 x, B 中剩余长度为 y,不失一般性地设 x≤y,现在要求的就是 ΣC(x,i) X C(y,i) = ΣC(x,x-i) X C(y,i) = C(x+y,x),这部分也是 O(|s|*|t|) 的。
需要注意的点是,两个数组如果都开ll,会MLE,所以开两个int的,在运算的时候用1ll转成ll,模完之后再变回int
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e3+10;
const int p=1e9+7;
string a,b;
int n,m;
int s[maxn][maxn],f[maxn][maxn];
int main(){
cin>>a>>b;
n=a.size(),m=b.size();
a='#'+a,b='#'+b;
for(int i=0;i<=n+1;i++) s[i][0]=1;
for(int i=0;i<=m+1;i++) s[0][i]=1;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
//注意此处的两个dp公式
计算相同前缀
if(a[i]==b[j]) s[i][j]=(1ll*s[i-1][j]+s[i][j-1])%p;
else s[i][j]=(1ll*s[i-1][j]+s[i][j-1]-s[i-1][j-1]+p)%p;
}
}
for(int i=0;i<=n+1;i++) f[i][m+1]=1;
for(int i=0;i<=m+1;i++) f[n+1][i]=1;
for(int i=n;i;i--){
for(int j=m;j;j--){
//计算后缀种类数
f[i][j]=(1ll*f[i+1][j]+f[i][j+1])%p;
}
}
ll ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(a[i]<b[j]){
ans=(ans+1ll*s[i-1][j-1]*f[i+1][j+1]%p)%p;
}
}
}
cout<<ans<<endl;
return 0;
}