CodeForces 107 B.Basketball Team（水~）

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;
}