【牛客】D 扔硬币(组合数)

链接: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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章