【概率】FZU 2103 Bin & Jing in wonderland

点击打开链接

给出N种物品，取K个，并且 取出的K个中按照编号排序 前R大的物品编号都给出

求出现取K个中按编号排序后前R大的物品编号为输入的概率

先知道R个中最小的编号==tmp，把比tmp大的概率算出

再枚举tmp的个数（num[tmp]<=     <=sum）并且把编号小于tmp看成一种物品

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <deque>
#include <map>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
#pragma comment(linker, "/STACK:102400000,102400000")
typedef long long  LL;
const int MAXN = 1000200;
const int MAXM = 140000;
const int INF = 0x3f3f3f3f;
const int mod = 1000000007;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
double p[22],ans;
LL c[53][53];
int num[22];
int maxx;
void getc(int n)
{
    int sum=1;
    c[n][0]=1;
    for(int i=1;i<=n;i++)
        c[n][i]=c[n][i-1]*(n-i+1)/i;
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    //  freopen("out.txt", "w", stdout);
#endif
    int t,n,k,r;
    for(int i=0;i<=52;i++)
        getc(i);
    cin>>t;
    while(t--)
    {
        cler(num,0);
        ans=1.0;
        cin>>n>>k>>r;
        for(int i=1;i<=n;i++)
            cin>>p[i];
        maxx=INF;
        for(int i=0;i<r;i++)
        {
            int x;
            cin>>x;
            num[x]++;
            maxx=min(maxx,x);
        }
        int x=k;
        for(int i=maxx+1;i<=n;i++)
        {
            ans*=c[x][num[i]]*pow(p[i],num[i]);
            x-=num[i];
        }
        double pi=0,ans2=0;
        for(int i=1;i<maxx;i++)
            pi+=p[i];
        int sum=k-r+num[maxx];
        for(int i=num[maxx];i<=sum;i++)// r个物品中编号最小有i个
            ans2+=pow(p[maxx],i)*c[sum][i]*pow(pi,sum-i);
        printf("%.6lf\n",ans*ans2);
    }
    return 0;
}