C++---贪心---闯关（每日一道算法2022.12.30）

本文链接：https://blog.csdn.net/SRestia/article/details/128499198

这篇博客介绍了一个涉及关卡挑战和特殊关卡修改分数的综艺节目问题。通过贪心策略，先计算普通关卡总分，然后按特殊关卡分数降序处理，以最大化总分。算法思路简洁，主要涉及排序和动态规划思想。博客作者分享了代码实现，并提及自身因疫情未能及时更新的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：
某综艺频道推出了一个闯关活动。
活动一共包含 n 个关卡（编号 1∼n），其中 m 个关卡为特殊关卡。
每个关卡都有一个通关分数，其中第 i 个关卡的通关分数为 ai。
挑战者可以自由决定所有关卡的具体挑战顺序，并且每通过一个关卡就可以获得该关卡的通关分数。
值得注意的是，当挑战者即将挑战的关卡是特殊关卡时，如果挑战者当前已经获得的总分数大于该特殊关卡的通关分数，则挑战者可以对该关卡的通关分数进行一次修改，修改后的新分数不能小于原分数，也不能大于挑战者当前已经获得的总分数。

请你计算并输出挑战者通过所有关卡获得的总分数的最大可能值。

输入格式
第一行包含两个整数 n,m。
第二行包含 n 个整数 a1,a2,…,an，表示每个关卡的通过分数。
第三行包含 m 个整数 b1,b2,…,bm，表示每个特殊关卡的编号。

输出格式
一个整数，表示挑战者通过所有关卡获得的总分数的最大可能值。
保证最终答案不超过 263−1。

数据范围
前 4 个测试点满足 1≤n≤4。
所有测试点满足 1≤n,m≤100，m≤min(n,30)，1≤ai≤107，1≤bi≤n，bi 两两不同。

输出：
18

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;
typedef long long LL;
const int N = 110;

// n是总挑战数量，m是特殊挑战数量
// w是所有挑战的分数，id存储第几个关卡是特殊挑战，st判断是否为特殊关卡
int n, m;
int w[N], id[N];
bool st[N];

int main()
{
    // 读入所有关卡分数，标记特殊关卡
    cin >> n >> m;
    for (int i = 1; i<=n; i++) cin >> w[i];
    for (int i = 0; i<m; i++) {
        cin >> id[i];
        st[id[i]] = true;
    }
    
    // 根据推断的贪心，将特殊关卡降序排序
    sort(id, id+m, [&](int a, int b){
        return w[a] > w[b]; 
    });
    
    // 先将普通关卡全部加在一起
    LL res = 0;
    for (int i = 1; i<=n; i++) {
        if (!st[i]) res += w[i];
    }
    
    // 最后将特殊关卡从大到小依次进行加到res中或者将res翻倍，这里可以优化为O(n)，因为只有最大的特殊关卡可能是相加，其他都是翻倍
    for (int i = 0; i<m; i++) {
        res = max(res*2, res + w[id[i]]);
    }
    
    cout << res << endl;
    
    return 0;
}