CodeForces - 768F Barrels and boxes —— 组合数

题意：

有n个盛食物的箱子，m个盛酒的箱子，只有同种箱子能叠在一起，只有不同种的箱子能相邻摆放，问所有酒箱垒出的高度都大于h的概率

思路：

需要求出所有摆放的情况总数和满足条件的情况数

枚举酒箱的堆数i，即把m分成i个不为0的数，由隔板法，情况总数为C(m-1,i-1)

这时食物箱的摆放会出现三种情况

i-1堆时，情况总数为C(n-1,i-2)

i堆时，情况总数为C(n-1,i-1)

i+1堆时，情况总数为C(n-1,i)

每个i下两种箱子的情况总数相乘即为总的情况总数

满足高度大于h的情况数是C(m-(h-1)*i,i-1)，就相当于先在i堆中每堆放h-1个，然后将剩余的箱子数m-(h-1)*i分成i个不为0的数

最后求一下逆元相乘即可

有点坑的是只有1个食物或者只由1个酒的情况...

#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define max_ 200100
#define mod 1000000007
#define inf 0x3f3f3f3f
#define eps 1e-9
ll n,m,h;
ll jc[100100],ijc[100100];
ll fpow(ll a,ll b)
{
    ll ans=1;
    ll tmp=a%mod;
    while(b)
    {
        if(b&1)
        ans=ans*tmp%mod;
        tmp=tmp*tmp%mod;
        b/=2;
    }
    return ans;
}
void init()
{
    jc[0]=jc[1]=1;
    for(int i=2;i<=100000;i++)
    jc[i]=jc[i-1]*i%mod;
    ijc[100000]=fpow(jc[100000],mod-2);
    for(int i=100000-1;i>=0;i--)
    ijc[i]=ijc[i+1]*(i+1)%mod;
}
ll C(int n,int m)
{
    if(m<0||m>n)
    return 0;
    return (jc[n]*ijc[m]%mod*ijc[n-m]%mod);
}
int main(int argc, char const *argv[]) {
    init();
    scanf("%lld%lld%lld",&n,&m,&h);
    if(n==0)
    {
        if(m>h)
        printf("1\n");
        else
        printf("0\n" );
        return 0;
    }
    if(m==0)
    {
        printf("1\n");
        return 0;
    }
    ll ans=0,tmp=0;
    ll w=m/(h+1);
    for(int i=1;i<=m;i++)
    {
        ll wine=C(m-1,i-1);
        ll food=((C(n-1,i-2)+2*C(n-1,i-1))%mod+C(n-1,i))%mod;
        tmp=(tmp+food*wine%mod)%mod;
        if(i<=w)
        {
            ll d=m-h*i;
            ans=(ans+food*C(d-1,i-1)%mod)%mod;
        }
    }
    printf("%lld\n",ans*fpow(tmp,mod-2)%mod);
    return 0;
}