【概率dp】2021年度训练联盟热身训练赛第三场 G.Research Productivity Index

Research Productivity Index

题目大意

给一堆论文和每个通过的概率 $p_i$，给定一个指数的计算方法，求指数的最大期望值。

解题思路

首先要明确要求的是什么！！！

求的是每种指数的最大期望。选择的每种论文数对应一种指数的期望，如选择 $1$ 篇论文，就求通过 $1$ 篇论文对应的期望；选择 $2$ 篇论文，要考虑通过 $1,2$ 篇的情况，以此类推。
对当前的选择的论文数，期望是通过的不同数量的论文对应的指数 $\times$ 对应的概率的和。

$dp[i][j]$ 维护前 $i$ 个论文通过 $j$ 篇的概率。

状态转移方程：
$$dp[i][j]=(dp[i-1][j]\ast (1.0-p[i]))+dp[i-1][j-1]\ast p[i];$$
最终求最大值，处理的时候一开始先对 $p$ 从大到小排序，之后枚举选择的论文数，求对应的期望即可。

参考代码

#include<stdio.h>
#include<iostream>
#include<vector>
#include<cstring>
#include<cstdio>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<deque>
#include<map>
#include<unordered_map>
#include<set>
#include<stack>
//#define LOCAL  //提交时一定注释
#define VI vector<int>
#define eps 1e-6
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
using namespace std;
typedef long long LL;
typedef double db;
const int inf = 0x3f3f3f3f;
const LL INF = 1e18;
const int N = 1e3 + 10;
#define ls rt << 1
#define rs rt << 1 | 1
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
inline int readint() {int x; scanf("%d", &x); return x;}

LL gcd(LL a, LL b) {
    return b == 0 ? a : gcd(b, a % b);
}

db dp[N][N];
db p[N];

db index(db n, db m) {
    return pow(m, m / n);
}
bool cmp(db x, db y) {
    return x > y;
}
int main() {
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
//   freopen("output.txt", "w", stdout);
#endif
    dp[0][0] = 1.0;
    int n = readint();
    for(int i = 1; i <= n; ++i) {
        p[i] = 1.0 * readint() / 100;
    }
    sort(p + 1, p + n + 1, cmp);
    db ans = 0;
    for(int i = 1; i <= n; ++i) {
        db tmp = 0;
        dp[i][0] = dp[i - 1][0] * (1 - p[i]);
        for(int j = 1; j <= i; j++) {
            dp[i][j] = (dp[i - 1][j] * (1.0 - p[i])) + dp[i - 1][j - 1] * p[i];
            tmp += dp[i][j] * index(i, j);
        }
        ans = max(ans, tmp);
    }
    printf("%.9f", ans);
    return 0;
}