“蔚来杯“2022牛客暑期多校训练营4 A题

Task Computing

题目链接

标签：数学 dp

题意：

• 有n个任务，每个任务有两种属性，w和p
• 从中选取m个任务，使下面这个式子结果最大
• 注意题目中给出的是w和q，需要用到的p=q/1000
  

思路：

下面是我听出题人讲的思路，此处的R_x和R_y都是总收益，比较其中间两项即可

由此我们可以得到比较关系
x.w * (1.0 - y.q) > y.w * (1.0 - x.q)
刚听到这里我以为本题就结束了，选前m项按公式一代就行，但这样很明显是错的
这里排序出来的实际意义是按此顺序从小到大选择能得到的收益最大
显然选择的任务情况相同时，最佳收益的顺序是上面排序得到数列的子序列
排序后我们在对任务进行dp
dp[i][j表示从后往前选,当前i个任务时已选j个任务
若选择接下第i个任务 dp[i][j] = dp[i + 1][j - 1] * a[i].q + a[i].w
不选dp[i][j] = dp[i+1][j]
最终dp[1][m]就是即为答案
这题对数学和思维都挺有要求，是到好题

代码：

#define fst std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout << std::fixed << std::setprecision(20)
#define le "\n"
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 50;
const int mod = 1e9 + 7;
double dp[N][30];

struct Node {
    double w, q;
};

bool cmp(Node x, Node y) {
    return x.w * (1.0 - y.q) > y.w * (1.0 - x.q);
}

int main() {
    fst;
    int n, m;
    cin >> n >> m;
    vector<Node> a(n);
    for (int i = 0; i < n; i++) cin >> a[i].w;
    for (int i = 0; i < n; i++) cin >> a[i].q, a[i].q /= 10000;
    sort(a.begin(), a.end(), cmp);
    for (int i = n - 1; i >= 0; i--) {
        for (int j = 1; j <= m; j++) {
            dp[i][j] = max(dp[i + 1][j], dp[i + 1][j - 1] * a[i].q + a[i].w);
        }
    }
    cout << dp[0][m] << le;
    return 0;
}