【算法模板】巧用优先队列

例题：找到和最大的长度为 K 的子序列

以力扣面试题 2099题为例：

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

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

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

---

示例 1:
输入：nums = [2,1,3,3], k = 2
输出：[3,3]
解释：
子序列有最大和：3 + 3 = 6 。

示例 2：
输入：nums = [-1,-2,3,4], k = 3
输出：[-1,3,4]
解释：
子序列有最大和：-1 + 3 + 4 = 6 。

示例 3：
输入：nums = [3,4,3,3], k = 2
输出：[3,4]
解释：
子序列有最大和：3 + 4 = 7 。
另一个可行的子序列为 [4, 3] 。

class Solution {
    public int[] maxSubsequence(int[] nums, int k) {
        int[] res = new int[k];
        PriorityQueue<Pair> queue = new PriorityQueue<>(k, (a, b) -> a.value - b.value);

        for (int i = 0; i < k; i++) {
            queue.add(new Pair(i, nums[i]));
        }

        for (int i = k; i < nums.length; i++) {
            if (queue.peek().value > nums[i]) {
                continue;
            }

            queue.poll();
            queue.add(new Pair(i, nums[i]));
        }

        PriorityQueue<Pair> queue2 = new PriorityQueue<>(k, (a, b) -> a.index - b.index);
        while (!queue.isEmpty()) {
            queue2.add(queue.poll());
        }

        int idx = 0;
        while (!queue2.isEmpty()) {
            res[idx] = queue2.poll().value;
            idx++;
        }

        return res;
    }

    class Pair {
        int index;
        int value;

        public Pair(int index, int value) {
            this.index = index;
            this.value = value;
        }
    }
}