题目链接:点击打开链接
题意:
给定n m h
表示有m个部门,有个人如今在部门h
以下m个数字表示每一个部门的人数。(包含他自己)
在这些人中随机挑选n个人,问挑出的人中存在和这个人同部门的概率是多少。
这个人一定在挑出的n个人中。
反向思考。答案是 1 - 不可能概率
不可能概率 = C(n-1, sum-1-a[h]) / C(n-1, sum-1)
发现2个组合数的分母部分同样,所以仅仅须要把2个组合数的分子部分相除就可以。
#include <cstdio> #include <iostream> #include <algorithm> #include <string.h> #include <map> #include <set> using namespace std; #define N 10010 int n, m, h, a[N]; void solve(){ int sum = 0; for(int i = 1; i <= m; i++) scanf("%d",&a[i]), sum += a[i]; if(sum < n){ puts("-1");return ; } n--; sum--; a[h]--; if(sum - a[h] < n){puts("1");return;} double ans = 1.0; double x = sum-a[h], y = sum; for(int i = 1; i <= n; i++) { ans *= x / y; x--; y--; } printf("%.10f ", 1.0 - ans); } int main(){ while(~scanf("%d %d %d",&n,&m,&h)){ solve(); } return 0; }