CodeForces 107 B.Basketball Team（水~）

最新推荐文章于 2022-09-10 15:07:35 发布

v5zsq

最新推荐文章于 2022-09-10 15:07:35 发布

阅读量471

点赞数

分类专栏： Code Forces 水题

本文链接：https://blog.csdn.net/V5ZSQ/article/details/78938945

版权

Code Forces 同时被 2 个专栏收录

606 篇文章

订阅专栏

水题

562 篇文章

订阅专栏

Description

有 $m$ 个团队，第 $i$ 个团队中有 $a_i$ 个人，现在要从这些人中选出 $n$ 个人组成一队，已知第 $h$ 个团队中有一个人一定被选，问最终队伍中有不少于两个来自第 $h$ 个团队的人的概率

Input

第一行三个整数 $n,m,h$ ，之后输入 $m$ 个整数 $a_i$ 表示第 $i$ 个团队的人数 $(1\le n\le 100,1\le m\le 1000,1\le h\le m,1\le a_i\le 100)$

Output

如果人数不够组队则输出 $-1$ ，否则输出组成满足条件队伍的概率

Sample Input

3 2 1
2 2

Sample Output

0.666667

Solution

令 $s=\sum\limits_{i=1}^ma_i$

如果 $s<n$ 显然输出 $-1$

如果 $s-a_h<n-1$ 则必然需要从第 $h$ 个团队中选取至少两个人，答案是 $1$

否则 $ans=\frac{C_{s-a_h}^{n-1}}{C_{s-1}^{n-1}}=\prod\limits_{i=1}^{n-1}\frac{s-a_h-n+1+i}{s-n+i}$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int main()
{
    int n,m,h;
    while(~scanf("%d%d%d",&n,&m,&h))
    {
        int s=0,ah;
        for(int i=1;i<=m;i++)
        {
            int a;
            scanf("%d",&a);
            if(i==h)ah=a;
            s+=a;
        }
        s--;n--;ah--;
        if(s<n)printf("-1\n");
        else if(s-ah<n)printf("1\n");
        else 
        {
            double ans=1;
            for(int i=1;i<=n;i++)ans=ans*(s-ah-n+i)/(s-n+i);
            ans=1.0-ans;
            printf("%.12f\n",ans);
        }

    } 
    return 0;
}