蓝桥杯2024年第十五届省赛真题-爬山

题目链接：

思路：

策略：每次都对最高的山使用魔法

①我们用大根堆存所有山的高度，每次取根节点，就是最大值。

②当两种魔法使用次数不为0，对取出的最大山的高度 t 使用魔法， 第一种魔法将高度变成 x， 第二种魔法将高度变成 y。 对 x， y比较，谁更小就用那种魔法，该魔法使用次数-1，将新的高度添加到大根堆中。

③当第二种魔法使用次数为0，将 sqrt(t) 添加到大根堆中，第二种魔法使用次数 -1。当第一种魔法使用次数为0，将 t/ 2 添加到大根堆中，第一种魔法使用次数 -1。

代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N  = 1000010;

int n, P, Q;
//大根堆
priority_queue<int, vector<int>, less<int>> q;
//答案
int ans;

signed main(){
    cin >> n >> P >> Q;
    for(int i = 1; i <= n; i++){
        int x; cin >> x;
        //从大到小排序
        q.push(x);
    }
    // //找到最小体力值
    while(P != 0 || Q != 0){
        //取出堆最大值
        auto t = q.top();
        //在堆中去掉最大值
        q.pop();
        //P Q 不为0
        if(P != 0 && Q != 0){
            int a = sqrt(t), b = t/2;
            if(a < b){
                q.push(a);
                P--;
            }
            else if(b < a){
                q.push(b);
                Q--;
            }
        }
        else if(P == 0 && Q != 0){
            q.push(t/2);
            Q--;
        }
        else if(P != 0 && Q == 0){
            q.push(sqrt(t));
            P--;
        }
    }
    

    while(q.size()){
        auto x = q.top();
        q.pop();
        ans += x;
    }
    
    cout << ans << endl;
    return 0;
}