20230501 - 栈与队列3 | 239. 滑动窗口最大值、347. 前 K 个高频元素

1、239. 滑动窗口最大值

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;
    }
}

 class MyQueue{
        Deque<Integer> deque = new LinkedList<>();
        //弹出元素时，比较当前要弹出的数值是否等于队列出口的数值，如果相等则弹出
        //同时判断队列当前是否为空
        void poll(int val){
            if(!deque.isEmpty() && val == deque.peek()){
                deque.poll();
            }
        }
        //添加元素时，如果要添加的元素大于入口处的元素，就将入口元素弹出
        //保证队列元素单调递减
        //比如此时队列元素3，1 2将要入队，比1大，所以1弹出，此时队列：3，2
        void add(int val){
            while(!deque.isEmpty()&&val>deque.getLast()){
                deque.removeLast();
            }
            deque.add(val);
        }
        //队列队顶元素始终为最大值
        int peek(){
            return deque.peek();
        }
    }

2、347. 前 K 个高频元素

class Solution {
    //首先使用一个hashmap记录key，value，然后使用大顶堆进行记录
    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer,Integer> map  = new HashMap<>();
        for(int num:nums){
             map.put(num,map.getOrDefault(num,0)+1); //记数
        }
        //定义一个优先级队列，大顶堆
        PriorityQueue<int[]> queue = new PriorityQueue<>((o1,o2)->{return o2[1]-o1[1];});
        //遍历map进行优先级的填充
        for(Map.Entry<Integer,Integer> entry:map.entrySet()){
            queue.add(new int[]{entry.getKey(),entry.getValue()});
        }
        int[] ans = new int[k];
        for(int i=0;i<k;i++){
            ans[i] = queue.poll()[0];
        }
        return ans;

    }
}