题意:
有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;
}