代码随想录算法训练营第十三天｜239. 滑动窗口最大值、347.前 K 个高频元素

代码随想录第十三天｜239. 滑动窗口最大值、347.前 K 个高频元素

239. 滑动窗口最大值

题目链接

问题简述：求出长度为k的滑动窗口中最大的元素，并存入n-k+1长度的数组中。

思考：暴力解法比较容易，但会超时。学习了单调队列的方法，主要是对deque不是很熟悉，depue双端队列，对队列的头first和尾last都可以进行加入删除查看值的操作，十分的方便。

算法思路：创建一个双端队列作为单调队列，队列存储元素下标。单调队列保证队列从头到尾数字从大到小，不断遍历数组元素，并将下标放入队列。当队头元素超出滑动窗口范围时，直接出队，不断删除队尾比新加入元素小的元素，保证队列从头到尾数字从大到小，最后将每次的队头加入数组。

时间复杂度 ：O(n) （每个元素最多被加入或推出一次）

空间复杂度 ：O(n)

import java.util.ArrayDeque;

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
      //定义一个双端队列作为单调队列，队列存储元素下标
        ArrayDeque<Integer> deque = new ArrayDeque<>();
      //定义输出的数组
        int[] res = new int[nums.length - k + 1];
      //用于遍历数组
        int idx = 0;
      //遍历nums元素
        for (int i = 0; i < nums.length; i++) {
            //如果队头元素小于等于i - k，说明该元素已经超出了k个以内的范围，或者其是k个范围内的第一个元素，直接将其推出（因为遍历到nums[i]时，本次循环就要添加nums[i]了，所以即使当前元素是k个范围内的第一个元素，也要推出）
            while (!deque.isEmpty() && deque.peekFirst() <= i - k){
                deque.removeFirst();
            }

            //不断删除队尾比新加入元素小的元素，保证队列从头到尾数字从大到小
            while (!deque.isEmpty() && nums[i] > nums[deque.peekLast()]){
                deque.removeLast();
            }
          
            //添加新元素的下标
            deque.offer(i);

            // 因为单调，当i增长到符合第一个k范围的时候，每滑动一步都将队列头节点放入结果就行了
            if(i >= k - 1){
                res[idx++] = nums[deque.peek()];
            }
        }
        return res;
    }
}

347.前 K 个高频元素

题目链接

问题简述：获得数组中前k个高频的元素。

思考：想到用map。然后学到了一些java语法：

• (pair1, pair2) -> pair1[1] - pair2[1]: lambda 表达式，用于比较两个整型数组。lambda 表达式定义了比较器，指定了数组按照第二个元素（索引为 1 的元素）升序排序。具体来说，pair1[1] - pair2[1] 表示比较两个数组的第二个元素的大小差值，如果差值为负数，则说明 pair1 应该排在 pair2 前面；如果差值为正数，则说明 pair1 应该排在 pair2 后面；如果差值为零，则说明它们相等，排序不变。

算法思路：用map遍历数组，key为数组元素，value为元素出现次数。然后不断将map元素传入优先级队列，取队列的前k个元素的value存入数组。

时间复杂度 ：O(n)

空间复杂度 ：O(n)

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
      //将元素传入map，统计出现次数
        for (int num : nums) {
            //getOrDefault用于获取指定键对应的值，如果键不存在，则返回指定的默认值0
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
      //创建优先级队列，元素整数型数组，优先级队列会根据数组中的第二个元素的大小自动进行排序
        PriorityQueue<int[]> pq = new PriorityQueue<>((pair1, pair2) -> pair1[1] - pair2[1]);
      //遍历map，同时遍历键和值，可以使用entrySet()方法来获取Map.Entry对象的集合
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
          //分别将map的key和value存入int型数组
            pq.add(new int[]{entry.getKey(), entry.getValue()});
          //取前k个
            if (pq.size() > k){
                pq.poll();
            }
        }
      
        int[] a = new int[k];
        for (int i = k - 1; i >= 0; i--) {
          //将key存入数组
            a[i] = pq.poll()[0];
        }
        return a;
    }
}

感想

这两个题写完了很久才写文档，发现忘记的很快，还是得保持记录。主要是对deque和PriorityQueue的不熟悉。