题目:
题意:
每个人都有固定的概率投出“好”和“坏”
求在
n
n
n个人内选出
k
k
k个使得两者数量相等的概率最大是多少
分析:
因为概率的计算规则,所以我们不难发现我们选择的
k
k
k个要尽量接近
1
1
1或者
0
0
0,具体证明不会的啦,大家感性理解一波
在排序后,我们如何计算概率呢
设
f
i
,
j
f_{i,j}
fi,j表示在前
i
i
i个人内投出了
j
j
j个好,对于新加入的人,要么是好,要么是坏,所以有方程:
f
i
,
j
+
=
{
f
i
−
1
,
j
−
1
∗
x
i
f
i
−
1
,
j
∗
(
1
−
x
i
)
f_{i,j}+=\left\{\begin{matrix}f_{i-1,j-1}*x_i\\f_{i-1,j}*(1-x_i)\end{matrix}\right.
fi,j+={fi−1,j−1∗xifi−1,j∗(1−xi)
可是我们无法确定到底前面选几个,后面选几个,所以我们再从后面做一次
d
p
dp
dp,最后用组合起来就好了
代码:
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#include<vector>
#define LZX Mu
#define LL long long
using namespace std;
inline LL read() {
LL d=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
return d*f;
}
double f[2005][2005],g[2005][2005],x[2005];
bool cmp(double a,double b) {return a>b;}
int main()
{
freopen("vote.in","r",stdin);
freopen("vote.out","w",stdout);
int n=read(),k=read();
for(int i=1;i<=n;i++) cin>>x[i];
sort(x+1,x+1+n);
f[0][0]=g[0][0]=1;
for(int i=1;i<=k;i++) for(int j=0;j<=i;j++) {f[i][j]=f[i-1][j]*(1-x[i]);if(j) f[i][j]+=f[i-1][j-1]*x[i];}
sort(x+1,x+1+n,cmp);
for(int i=1;i<=k;i++) for(int j=0;j<=i;j++) {g[i][j]=g[i-1][j]*(1-x[i]);if(j) g[i][j]+=g[i-1][j-1]*x[i];}
double ans=0;
for(int i=0;i<=k;i++)
{
double sum=0;
for(int j=0;j<=k/2;j++) sum+=f[i][j]*g[k-i][k/2-j];
ans=max(ans,sum);
}
printf("%.6lf",ans);
return 0;
}