洛谷B3872 [GESP202309 五级] 巧夺大奖-普及/提高-

题目描述

小明参加了一个巧夺大奖的游戏节目。主持人宣布了游戏规则：

1. 游戏分为 $n$ 个时间段，参加者每个时间段可以选择一个小游戏。

2. 游戏中共有 $n$ 个小游戏可供选择。

3. 每个小游戏有规定的时限和奖励。对于第 $i$ 个小游戏，参加者必须在第 $T_i$ 个时间段结束前完成才能得到奖励 $R_i$。

小明发现，这些小游戏都很简单，不管选择哪个小游戏，他都能在一个时间段内完成。关键问题在于，如何安排每个时间段分别选择哪个小游戏，才能使得总奖励最高？

输入格式

输入第一行，包含一个正整数 $n$。$n$ 既是游戏时间段的个数，也是小游戏的个数。约定 $1\le n\le 500$。

输入第二行，包含 $n$ 个正整数。第 $i$ 个正整数为 $T_i$，即第 $i$ 个小游戏的完成期限。约定 $1\le T_i\le n$。

输入第三行，包含 $n$ 个正整数。第 $i$ 个正整数为 $R_i$，即第 $i$ 个小游戏的完成奖励。约定 $1\le R_i\le 1000$。

输出格式

输出一行，包含一个正整数 $C$，为最高可获得的奖励。

输入输出样例 #1

输入 #1

7
4 2 4 3 1 4 6
70 60 50 40 30 20 10

输出 #1

230

说明/提示

样例解释 1

$7$ 个时间段可分别安排完成第 4、2、3、1、6、7、5 个小游戏，其中第 4、2、3、1、7 个小游戏在期限内完成。因此，可以获得总计 $40+60+50+70+10=230$ 的奖励。

solution

贪心：从最后一个时间段开始，选择结束时间在本时间段之后的、还未选择的、效益最高的小游戏完成

代码

#include <iostream>
#include "bit"
#include "vector"
#include "unordered_set"
#include "set"
#include "queue"
#include "algorithm"      

using namespace std;

/*
 * 从最后一个时间段开始，选择结束时间在本时间段 i 之后的还未选择的效益最高的小游戏
 */

int T[501], R[501], id[501], n, m, x;

int main() {
    cin >> n;
    for (int i = 0; i < n; i++) cin >> T[i];
    for (int i = 0; i < n; i++) {
        cin >> R[i];
        id[i] = i;
    }
    // 按照时间段降序排列
    sort(id, id + n, [&](int i, int j) {
        return T[i] > T[j];
    });

    // for (int i = 0; i < n; i++) cout << id[i];

    int s = 0;
    priority_queue<int> q;
    int j = 0;
    for (int i = 0; i < n; i++) {
        while (j < n && T[id[j]] >= n - i){
            // cout << R[id[j]] << " " << j << endl;
            q.push(R[id[j]]);
            j++;
        }

        if(!q.empty()){
            s += q.top();
            q.pop();
        }
    }

    cout << s;
    return 0;
}

结果