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

 1 239. 滑动窗口最大值

题目：给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

提示：

• 1 <= nums.length <= 105
• -104 <= nums[i] <= 104
• 1 <= k <= nums.length

思路：

（1）最初思路：用队列模拟窗口，依次取k个值比较得出最大值，后续弹出队首元素，添加队尾元素，比较最大值，循环结束。

（2）进阶思路【看教程后】： 窗口移动中维护单调递减队列，确保留在队列中队首元素为当前窗口中的最大值。

注意：进阶思路中需要自定义队列，各方法需要重写。

Java实现

class MyQueue{
    Deque<Integer> deque = new LinkedList<>();
    
    // 弹出元素时，比较当前要弹出元素是否等于队列出口的数值，如果相等则弹出
    // 同时判断队列当前是否为空
    void poll(int val){
        if(!deque.isEmpty() && val==deque.peek()){
            deque.poll();
        }
    }

    // 添加元素时，如果要添加元素大于队列入口处元素，将入口元素弹出
    // 保证队列单调递减
    void add(int val){
        while(!deque.isEmpty() && val>deque.getLast()){
            deque.removeLast();
        }
        deque.add(val);
    }

    // 队列顶元素始终为最大值
    int peek(){
        return deque.peek();
    }
}
class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length == 1){
            return nums;
        }
        int len = nums.length - k + 1;
        // 存放结果元素的数组
        int[] res = new int[len];
        int num = 0;
        // 自定义队列
        MyQueue myQueue = new MyQueue();
        // 将前K个元素放入队列中
        for(int i = 0;i<k;i++){
            myQueue.add(nums[i]);
        }
        res[num++] = myQueue.peek();
        for(int i=k;i<nums.length;i++){
            // 滑动窗口移除最前面的元素
            myQueue.poll(nums[i-k]);
            // 滑动窗口加入最后面的元素
            myQueue.add(nums[i]);
            // 记录对应的最大值
            res[num++] = myQueue.peek();
        }
        return res;

    }
}

2 347. 前 K 个高频元素

题目：给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

提示：

• 1 <= nums.length <= 105
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

思路：

（1）最初思路：用两个表分别记录元素值和元素出现次数，根据出现次数表排序后，对应将元素值表调整顺序，依次输出前K个值。

（2）进阶思路：用map记录元素值和出现次数，用大顶推排序后输出前K个值。【不太熟悉map，代码部分可以说是打了一遍勉强AC】

注意：熟悉Java的各种数据类型

Java实现

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer,Integer> map = new HashMap<>(); // key为数组元素值，val为对应出现次数
        for(int num:nums){
            map.put(num,map.getOrDefault(num,0)+1);
        }
        // 在优先队列中存储二原组（num，cnt），cnt表示元素值num在数组中出现的次数
        // 大顶推
        PriorityQueue<int[]> pq = new PriorityQueue<>((pair1,pair2)->pair2[1]-pair1[1]);
        for(Map.Entry<Integer,Integer> entry:map.entrySet()){
            pq.add(new int[]{entry.getKey(),entry.getValue()});
        }
        int[] ans = new int[k];
        for(int i=0;i<k;i++){
            ans[i] = pq.poll()[0];
        }
        return ans;
    }
}