【代码随想录Day13】239.滑动窗口最大值、347.前K个高频元素

joyboy0405

已于 2023-05-22 11:47:42 修改

阅读量214

点赞数

文章标签： leetcode 算法数据结构

于 2023-05-22 11:44:48 首次发布

本文链接：https://blog.csdn.net/weixin_38550348/article/details/130803273

版权

文章介绍了两种使用数据结构解决算法问题的方法：一是利用单调队列找到数组中滑动窗口的最大值；二是通过大顶堆和小顶堆分别实现寻找数组中出现频率最高的前K个元素。在滑动窗口最大值问题中，使用单调队列保证了O(1)的删除和插入操作；在高频元素问题中，大顶堆直接选出频率最高的元素，而小顶堆则在堆满后用于更新频率较低的元素。

摘要由CSDN通过智能技术生成

Day13

239.滑动窗口最大值

	本题需要构造单调队列来完成。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        ArrayDeque<Integer> deque = new ArrayDeque<>();
        int n = nums.length;
        int[] ans = new int[n - k + 1];
        int idx = 0;
        for (int i = 0; i < n; i++) {
            while (!deque.isEmpty() && deque.peek() < i - k + 1) {
                deque.poll();
            }

            while (!deque.isEmpty() && nums[deque.getLast()] < nums[i]) {
                deque.removeLast();
            }
            deque.offer(i);
            if (i >= k - 1) {
                ans[idx++] = nums[deque.peek()];
            }
        }
        return ans;
    }
}

347.前K个高频元素

	本题需要使用优先队列。

基于大顶堆

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>((pair1, pair2) -> pair2[1] - pair1[1]);
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
        }
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
        }
        int[] ans = new int[k];
        for (int i = 0; i < k; i++) {
            ans[i] = priorityQueue.poll()[0];
        }
        return ans;
    }
}

基于小顶堆

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>((pair1, pair2) -> pair1[1] - pair2[1]);
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
        }
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (priorityQueue.size() < k) {
                priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
            } else {
                if (entry.getValue() > priorityQueue.peek()[1]) {
                    priorityQueue.poll();
                    priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
                }
            }
        }
        int[] ans = new int[k];
        for (int i = 0; i < k; i++) {
            ans[i] = priorityQueue.poll()[0];
        }
        return ans;
    }
}