2021牛客暑期多校训练营5 D-Double Strings(dp+组合数)

D-Double Strings
f i , j f_{i,j} fi,j表示a中前i个字符,b中前j个字符相同子序列的数量,容斥转移

f i , j = f i − 1 , j + f i , j − 1 − f i − 1 , j − 1 + { ( 1 + f i − 1 , j − 1 ) [ a i = a j ] } f_{i,j}=f_{i-1,j}+f_{i,j-1}-f_{i-1,j-1}+\{(1+f_{i-1,j-1})[a_i=a_j]\} fi,j=fi1,j+fi,j1fi1,j1+{(1+fi1,j1)[ai=aj]}

g i , j g_{i,j} gi,j表示a中前i个字符,b中前j个字符满足小于关系子序列的数量

类似相同子序列数量转移即可。
g i , j = g i − 1 , j + g i , j − 1 − g i − 1 , j − 1 + ( 1 + g i − 1 , j − 1 ) + { ( f i − 1 , j − 1 + 1 ) [ a i < a j ] } g_{i,j}=g_{i-1,j}+g_{i,j-1}-g_{i-1,j-1}+(1+g_{i-1,j-1})+\{(f_{i-1,j-1}+1)[a_i<a_j]\} gi,j=gi1,j+gi,j1gi1,j1+(1+gi1,j1)+{(fi1,j1+1)[ai<aj]}

Code1
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
template <class T=int> T rd()
{
    T res=0;T fg=1;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') fg=-1;ch=getchar();}
    while( isdigit(ch)) res=(res<<1)+(res<<3)+(ch^48),ch=getchar();
    return res*fg;
}
const int N=5010;
const ll mod=1000000007;
char a[N],b[N];
int f[N][N],g[N][N];
int n,m;
int main()
{
    scanf("%s%s",a+1,b+1);
    n=strlen(a+1);m=strlen(b+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            if(a[i]==b[j])
                f[i][j]=(1ll*f[i-1][j]+f[i][j-1]+1)%mod;
            else
                f[i][j]=(1ll*mod+f[i-1][j]+f[i][j-1]-f[i-1][j-1])%mod;
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            if(a[i]<b[j]) 
                g[i][j]=(1ll*g[i-1][j]+g[i][j-1]+1+f[i-1][j-1])%mod;
            else
                g[i][j]=(1ll*g[i-1][j]+g[i][j-1])%mod;
        }
        
    printf("%lld\n",g[n][m]);
}
Code2

首先求出相同子序列,然后枚举哪个位置不同,即第一个 a i < a j a_i<a_j ai<aj,然后利用下面公式加快计算。

∑ 0 ≤ i ≤ k ( n i ) ( m k − i ) = ( n + m k ) \sum_{0\leq i\leq k}\dbinom{n}{i}\dbinom{m}{k-i}=\dbinom{n+m}{k} 0ik(in)(kim)=(kn+m)

于是有
∑ 0 ≤ i ≤ m ( n i ) ( m i ) ∑ 0 ≤ i ≤ m ( n i ) ( m m − i ) = ( n + m m ) \sum_{0\leq i\leq m}\dbinom{n}{i}\dbinom{m}{i}\sum_{0\leq i\leq m}\dbinom{n}{i}\dbinom{m}{m-i}=\dbinom{n+m}{m} 0im(in)(im)0im(in)(mim)=(mn+m)

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
template <class T=int> T rd()
{
    T res=0;T fg=1;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') fg=-1;ch=getchar();}
    while( isdigit(ch)) res=(res<<1)+(res<<3)+(ch^48),ch=getchar();
    return res*fg;
}
const int N=5010;
const int mod=1000000007;
char a[N],b[N];
int f[N][N];
int n,m;
int fac[N<<1],inv[N<<1];
ll qmi(ll a,ll b){ll v=1;while(b){if(b&1) v=v*a%mod;a=a*a%mod;b>>=1;}return v;}
void init()
{
    fac[0]=1;
    for(int i=1;i<=10000;i++) fac[i]=1ll*fac[i-1]*i%mod;
    inv[10000]=qmi(fac[10000],mod-2);
    for(int i=9999;i>=0;i--) inv[i]=1ll*inv[i+1]*(i+1)%mod;
}
int C(int n,int m){return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;}
int main()
{
    init();
    scanf("%s%s",a+1,b+1);
    n=strlen(a+1);m=strlen(b+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            if(a[i]==b[j])
                f[i][j]=(1ll*f[i-1][j]+f[i][j-1]+1)%mod;
            else
                f[i][j]=(1ll*mod+f[i-1][j]+f[i][j-1]-f[i-1][j-1])%mod;
        }
    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*(f[i-1][j-1]+1)*C(n-i+m-j,n-i)%mod)%mod;
            
            
    printf("%lld\n",ans);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值