CF 107B Basketball Team [排列组合]

类似高中的盒子取球，关键理解题意

问题转化为，从m个盒子中取n个球，在取了h盒中的一个球的条件下，h盒还取了别的球的概率

可以这样求得

sum表示m个盒子中总共有多少个球

从sum-1中取n-1个球的情况数是总情况数

减去从sum-m[h]取n-1个球的情况数

即是要求的情况数

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
int m[1111];
int main(){
#ifndef ONLINE_JUDGE
	freopen("/home/rainto96/in.txt","r",stdin);
#endif

	int n,mm,h;cin>>n>>mm>>h;
	int sum=0;
	for(int i=1;i<=mm;i++){
		cin>>m[i];
		sum+=m[i];
	}
	if(sum<n)
		cout<<-1<<endl;
	else{
		double ans=1;
		for(int i=0;i<n-1;i++){
			ans*=(sum-m[h]-n+2+i)*1.0/(sum-n+1+i);
		}
		ans=1-ans;
		cout<<ans<<endl;
	}
}