标签搜索

目 录CONTENT

文章目录

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

陈铭
2022-10-21 / 0 评论 / 1 点赞 / 241 阅读 / 370 字 / 正在检测是否收录...

例题:找到和最大的长度为 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;
        }
    }
}
1

评论区