cf 108b basketball team 组合数学

题意：

某大学有m个学院，每一个学院有si个人，现在要组建一支有n个人的篮球队，已经知道男主角在第h个学院，还有他先被球队录取了，现在学院在剩下的所有人中等概率的挑选出n-1个人

问：男主角想知道球队里面还有人和他来自同一个学院的概率，若不够人数，则输出-1

这道题很简单的组合数学题，但是要注意，计算的中间过程溢出。

 

#include<cstdio>

const int maxn=1010;
#define LL long long

int s[maxn];

int main()
{
    int n,m,h;
    scanf("%d%d%d",&n,&m,&h);
    int sum=0;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&s[i]);
        sum+=s[i];
    }
    if(sum<n)
    {
        printf("-1\n");
    }
    else if(s[h]==1)
    {
        printf("0\n");
    }
    else if(sum-s[h]<n-1)
    {
        printf("1\n");
    }
    else
    {
        s[h]--;
        sum--;
        double cnta=1;
        for(LL i=sum-s[h]-n+2;i<=sum-n+1;i++)
        {
            cnta*=i;
        }
        for(LL i=sum-s[h]+1;i<=sum;i++)
        {
            cnta=cnta/i;
        }
        printf("%.6f\n",1.0-(double)cnta);
    }

    return 0;
}