链接:https://ac.nowcoder.com/acm/contest/5758/D
来源:牛客网
题目描述
有n枚硬币,每枚硬币扔出来是正面和反面的概率各占50%。小明同时扔下了n枚硬币后,已知至少有m枚硬币是反面。请问恰好有k枚硬币是正面的概率是多少。
输入描述:
输入t,代表有t组数据。每组数据输入一个数n,m,k,代表有n枚硬币,抛出以后至少有m枚是反面的情况下,恰好有k个正面的概率。
(t<=1000,n<1e5,m<=1000,k<=n)
输出描述:
对于结果是p/q,输出分数取模1e9+7后的结果。
示例1
输入
1
10 3 5
输出
797520667
思路:由于已知部分硬币的方向。因此,扔n个硬币的情况要在2的n次幂的中去掉少于m个硬币是反面的情况。
对于k+m>n是不可能的,直接输出0.
其余情况:C(n,k)/[ 2^n - ∑(C(n,i)) ] (i<m)
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10;
const int inf=0x3f3f3f3f;
int mod=1e9+7;
typedef long long ll;
ll fact[N],infact[N];
ll ksm(ll a,ll n){
ll sum=1;
while(n){
if(n&1) sum=sum*a%mod;
a=a*a%mod;
n>>=1;
}
return sum;
}
void init(){
fact[0]=infact[0]=1;
for(int i=1;i<N;i++){
fact[i]=fact[i-1]*i%mod;//求逆元
infact[i]=infact[i-1]*ksm(i,mod-2)%mod;//求阶乘的逆元
}
}
ll C(ll a,ll b){//在a个里面取b个
return fact[a]%mod*infact[b]%mod*infact[a-b]%mod;
}
int main()
{
ll n,i,j,m,k,x,y,T;
init();
scanf("%lld",&T);
while(T--){
scanf("%lld %lld %lld",&n,&m,&k);
if(m+k>n){
puts("0");continue;
}
x=ksm(2,n);y=C(n,k);
for(i=0;i<m;i++){
x=(x-C(n,i)+mod)%mod;
}
printf("%lld\n",y*ksm(x,mod-2)%mod);
}
return 0;
}