Leetcode 2099. 找到和最大的长度为 K 的子序列 --- 堆解法

2099. 找到和最大的长度为 K 的子序列

给你一个整数数组 nums 和一个整数 k 。你需要找到 nums 中长度为 k 的 子序列 ，且这个子序列的 和最大 。

请你返回 任意 一个长度为 k 的整数子序列。

子序列 定义为从一个数组里删除一些元素后，不改变剩下元素的顺序得到的数组。

提示：

• 1 <= nums.length <= 1000
• -105 <= nums[i] <= 105
• 1 <= k <= nums.length

---

本题使用优先队列进行解题

具体思路：

• 自定义一个my_pair,功能与pair基本一致，但加入了小于号的重载[为后期的自定义数据类型加入优先队列做准备]
  • 自定义一个功能与pair完全一致的目的就是能够构建一个降序排列的大根堆
• pair中存储数据值+数据的idx，本题中不仅仅是找到最大的相邻k个数，还要能够按顺序排列进行返回，因此pair中加入idx用于后期对下标的排序

---

源码：

class Solution {
    class my_pair {  // 自定义类型
        public:
            int index;
            int val;
            bool operator<(const my_pair &b) const {
                return val < b.val;  // 大顶堆  --  降序排列
            }
            my_pair(){
        
            }
            my_pair(int idx, int v):index(idx), val(v){}  // 注意 需要构造函数
    };
public:
    vector<int> maxSubsequence(vector<int>& nums, int k) {
        priority_queue<my_pair> q;
        for(int i = 0; i < nums.size(); ++i) {   
        // 经过运算符的重载，自定义数据类型可以正确的加入优先队列，且顺序是按照val从大到小顺序排列
            q.push(my_pair(i, nums[i]));
        }
        vector<my_pair> tmp;   
        while(k--) {// 把优先队列中的前k个元素读出
            tmp.push_back(q.top());   // top获取完后需要弹出
            q.pop();
        }
        sort(tmp.begin(), tmp.end(), [](const my_pair& a, const my_pair& b){
           return a.index<b.index; // -> 这里这种方式就是将vector按照idx从小到大的顺序排列
        });  // sort函数的第三个参数本身就是一个比较函数，这里用了lambda的方法
// lambda--用于产生一个匿名小规模且作用域有限的函数，用于简化代码
        vector<int> ans;
        ans.clear();
        for(int i = 0; i < tmp.size(); ++i) {
            ans.push_back(tmp[i].val);   // 构造答案
        }
        return ans;、
    }
};