CodeForces 306 C.White, Black and White Again（组合数学）

v5zsq

于 2018-01-05 15:06:53 发布

阅读量357

点赞数

分类专栏： Code Forces 组合数学

本文链接：https://blog.csdn.net/V5ZSQ/article/details/78973105

版权

Code Forces 同时被 2 个专栏收录

606 篇文章 6 订阅

订阅专栏

组合数学

190 篇文章 1 订阅

订阅专栏

Description

给出 $w$ 个带编号黑球， $b$ 个带编号白球，要求把这些球放入 $n$ 个盒子里，每个盒子非空，且这些盒子可以被分成三段非空盒子集合，第一段和第三段全是黑球，第二段全是白球，问方案数

Input

三个整数 $n,w,b(3\le n\le 4000,2\le w\le 4000,1\le b\le 4000,w+b\ge n)$

Output

输出方案数，结果模 $10^9+9$

Sample Input

3 2 1

Sample Output

Solution

先不考虑黑白球编号，求出答案后乘上 $w!\cdot b!$ 即可，枚举白球占盒子数 $i$ ，那么黑球占盒子数为 $n-i$ ，白球占盒子区间有 $n-i-1$ 种可能，白球所占区间固定则黑球所占区间固定，剩下问题为将 $b$ 个白球分到 $i$ 个不同的盒子且每个盒子非空，将 $w$ 个白球分到 $n-i$ 个盒子里且每个盒子非空，由插板法知答案为 $w!\cdot b!\cdot \sum(n-i-1)\cdot C_{b-1}^{i-1}\cdot C_{w-1}^{n-i-1},1\le i\le b,2\le n-i\le w$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=4005;
#define mod 1000000009
int fact[maxn],inv[maxn];
void init(int n=4000)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int C(int n,int m)
{
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
    init();
    int n,w,b;
    while(~scanf("%d%d%d",&n,&w,&b))
    {
        int ans=0;
        for(int i=1;i<=b;i++)
            if(n-i<=w&&n-i>=2)
                ans=(ans+(ll)(n-i-1)*C(b-1,i-1)%mod*C(w-1,n-i-1)%mod)%mod;
        ans=(ll)ans*fact[w]%mod*fact[b]%mod;
        printf("%d\n",ans);
    }
    return 0;
}