代码跟学Day13【代码随想录】

Svetlana丶

已于 2023-06-20 10:31:14 修改

阅读量37

点赞数

分类专栏：算法跟学文章标签：算法 leetcode 数据结构 java

于 2023-06-19 10:57:51 首次发布

本文链接：https://blog.csdn.net/qq_37016176/article/details/131283166

版权

算法跟学专栏收录该内容

32 篇文章 0 订阅

订阅专栏

第五章栈与队列03

大纲

239 滑动窗口最大值
347 前K个高频元素
总结

`leetcode 239` 滑动窗口最大值

思路

单调队列经典问题

细节

注意单调队列的维护过程
- 进队列的元素需要从队列尾部开始依次比较找到属于自己的位置
- 出队列的元素需要判断是否与当前区间最前的元素相同

代码

class Solution {
    class CustomQueue {
        Deque<Integer> deque = new LinkedList<>();
        void poll(int val) {
            if (!deque.isEmpty() && val == deque.peek()) deque.poll();
        }
        void offer(int val) {
            while (!deque.isEmpty() && val > deque.getLast()) deque.pollLast();
            deque.offer(val);
        }
        int peek() {
            // System.out.println(deque.toString());
            return deque.peek();
        }
    }

    public int[] maxSlidingWindow(int[] nums, int k) {
        CustomQueue cq = new CustomQueue();
        int len = nums.length, index = 0;
        int[] res = new int[len - k + 1];
        for (int i = 0; i < k; i++) cq.offer(nums[i]);
        res[index++] = cq.peek();
        for (int i = k; i < len; i++) {
            cq.poll(nums[i - k]);
            cq.offer(nums[i]);
            res[index++] = cq.peek();
        }
        return res;
    }
}

复杂度

时间 O(n)
空间 O(n)

`leetcode 239` 前 K 个高频元素

思路

先遍历一次统计元素频次
再使用自定义比较的优先队列找出前k个高频元素

细节

注意优先队列实质上是堆在比较时需要自定义比较在这里全程是用hash的键来比较值的

代码

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums) map.put(i, map.getOrDefault(i, 0) + 1);
        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> map.get(a) - map.get(b));
        for (int i : map.keySet()) 
            if (queue.size() < k) queue.offer(i);
            else if (map.get(i) > map.get(queue.peek())) {
                queue.poll();
                queue.offer(i);
            }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) res[i] = queue.poll();
        return res;
    }
}

复杂度

时间 O(nlogk)
空间 O(n)

总结

栈与队列这一章节针对了栈和队列这两种经典数据结构的应用
- 栈
  - 括号匹配
  - 字符串去重
  - 逆波兰式
  - 简化路径
- 队列
  - 滑动窗口最大值单调队列
  - 前k个高频元素优先队列
注意在Java语言中栈和队列的实现
- 栈 Stack 队列 LinkedList PriorityQueue ArrayQueue