BJ模拟:计数(组合)

n1 n 1 A A n2 B B n3 C C n4 D D 排成一个序列,求有多少种排列方案使得排成的序列任意两个相邻字不相同(n1000)

题解:
fi f i 表示 A,B A , B 排成的所有序列有 i i 个不合法位置的方案,gi表示 C,D C , D 排成 i i 个序列,每个序列都合法的方案。显然答案为:

iA+B1j=iC+Dfigj(A+B+1iji)

考虑如何求出 f,g f , g
对于 f f 枚举A的几个空位插入N即可。
对于 g g 枚举有多少个序列是C D D <script type="math/tex" id="MathJax-Element-1796">D</script>多1,那么少1的也可确定。再枚举为0的个数,之后用组合数计算即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}

const int N=4e3+50, mod=1e9+7;
inline int add(int x,int y) {return (x+y>=mod) ? (x+y-mod) : (x+y);}
inline int dec(int x,int y) {return (x-y<0) ? (x-y+mod) : (x-y);}
inline int mul(int x,int y) {return (LL)x*y%mod;}
inline int power(int a,int b,int rs=1) {for(;b;b>>=1,a=mul(a,a)) if(b&1) rs=mul(rs,a); return rs;}
int A,B,C,D,fac[N],ifac[N],pw[N],f[N],g[N];
inline int C2(int a,int b) {if(a<b) return 0;return mul(fac[a], mul(ifac[b],ifac[a-b]));}
inline int C3(int a,int b,int c,int d) {return mul( mul(fac[a],ifac[b]), mul(ifac[c],ifac[d]));}
int main() {
    A=rd(), B=rd(), C=rd(), D=rd();
    fac[0]=1; pw[0]=1; 
    for(int i=1;i<N;i++) fac[i]=mul(fac[i-1],i);
    ifac[N-1]=power(fac[N-1],mod-2);
    for(int i=N-2;~i;i--) ifac[i]=mul(ifac[i+1],i+1);
    for(int i=1;i<N;i++) pw[i]=mul(pw[i-1],2);
    if(A&&B) {
        for(int k=1;k<A && k<=B;k++) 
            f[A+B-2*k-1]=add(f[A+B-2*k-1],mul(C2(A-1,k),C2(B-1,k-1)));
        for(int k=1;k<B && k<=A;k++)
            f[A+B-2*k-1]=add(f[A+B-2*k-1],mul(C2(B-1,k),C2(A-1,k-1)));    
        for(int k=1;k<=A && k<=B;k++)
            f[A+B-2*k]=add(f[A+B-2*k],mul(2, mul(C2(A-1,k-1),C2(B-1,k-1))));
    } else {
        if(A<B) swap(A,B);
        if(!A) f[0]=1;
        else f[A-1]=1;  
    }
    if(C&&D) {
        if(C<D) swap(C,D);
        for(int i=C-D;i<=C;i++) {
            int j=i-(C-D);
            if(j>D) break;
            for(int k=0;k<=C-i;++k) 
                g[i+j+k]=add(g[i+j+k], mul(mul(pw[k], C2(j+C-1,C-i-k)),C3(i+j+k,i,j,k)));   
        }
    } else {
        if(C<D) swap(C,D);
        if(!C) g[0]=1;
        else g[C]=1;
    }
    int ans=0;
    for(int i=0;i<=A+B;++i) {
        for(int j=i;j<=C+D;++j) {
            int rs=mul(mul(f[i],g[j]), C2(A+B+1-i, j-i));
            ans=add(ans,rs);
        }
    }
    cout<<ans;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值