中石油问题 K: ABBA(组合数学)

最新推荐文章于 2025-08-07 20:58:17 发布

林苏泽

最新推荐文章于 2025-08-07 20:58:17 发布

阅读量133

点赞数

CC 4.0 BY-SA版权

文章标签：算法机器学习 python 大数据人工智能

本文链接：https://blog.csdn.net/weixin_45911397/article/details/119341487

本文介绍了一种使用组合数学和容斥原理解决特定字符串计数问题的方法，通过计算合法的ABBA序列数量，并提供了一个有效的C++实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题 K: ABBA

题意：

给出n和m，n是表示AB的数量，m表示BA的数量，总长度为2*(n+m)然后分成(n+m)的长度为2的子串共包括n个AB,m个BA。问有多少种？对结果取模于\(1e9+7\)

分析：

显然我们需要用组合数学来解决问题，但是我们直接分析不好分析，那么就采用容斥原理，整体来看就是2*(n+m)种选出(n+m)个，然后我们需要减去不合法的。

ll n,m;
ll hoppy,lad;
ll equip[maxn],frown[maxn];
ll qpow(ll hoppy1,ll lad1){
    ll ans=1;
    while(lad1){
        if(lad1&1) ans=ans*hoppy1%mod;
        lad1>>=1;
        hoppy1=hoppy1*hoppy1%mod;
    }
    return ans;
}
void search(){  
    equip[0]=1;
    frown[0]=1;
    for(int i=1;i<=1e5;i++){
        equip[i]=equip[i-1]*i%mod;
        frown[i]=frown[i-1]*qpow(i,mod-2)%mod;
    }
}
ll C(ll hoppy,ll lad){
    return equip[hoppy]*frown[lad]%mod*frown[hoppy-lad]%mod;
}
 
void solve(){
    if(n==0&&m==0) {
        puts("1");
        return ;
    }
    ll ans=0;
    ll sum=(n+m)*2;
    ans=C(sum,n+m);
    if(m!=0)
    ans=(ans-C(sum,m-1)+mod)%mod;
    if(n!=0)
    ans=(ans-C(sum,n-1)+mod)%mod;
    cout<<ans<<endl;
}
 
int main(){
    search();
    while(~scanf("%lld%lld",&n,&m)) solve();
    return 0;
}