G. Graduation Guarantee-2022-2023 ACM-ICPC Nordic Collegiate Programming Contest (NCPC 2022)

Problem - G - Codeforces

题目大意：有n道题，要得k分，每道题做对+1分做错-1分，可以选择做或者不做，每道题做对的概率是pi,求能到达k分的最大概率是多少

思路：贪心+期望dp

先将概率从大到小进行排序，若该题做对的概率较小的话，不选这题得到k分的概率要比选这题的高，所以用dp[i][j]跑出前i题做对j题的概率是多少，然后遍历找出最大值。

#include<bits/stdc++.h>
using namespace std;
const int maxn=5007;
double a[maxn];
double f[maxn][maxn];//前i道题对j道题的概率
int n,k;
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    sort(a+1,a+1+n,greater<double>());//大的优先
    f[0][0]=1;
    for(int i=1;i<=n;i++){
        f[i][0]=(1-a[i])*f[i-1][0];
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            f[i][j]=a[i]*f[i-1][j-1]+(1-a[i])*f[i-1][j];
        }
    }
    double ans=0;
    for(int i=k;i<=n;i++){
        double sum=0;
        for(int j=i-(i-k)/2;j<=i;j++){//(i-k)/2是要拿i的分数最多能错多少道
            sum+=f[i][j];
        }
        ans=max(ans,sum);
    }
    cout<<ans<<"\n";
}