代码随想录Day13|239、347

239. 滑动窗口最大值

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

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        ArrayDeque<Integer> deque = new ArrayDeque<>();
        int n = nums.length;
        int[] res = new int[n- k + 1];
        int idx = 0;
        for(int i = 0; i < n; i++){
            // 1.队列头结点需要在[i-k+1,i]范围内，不符合则弹出
            while(!deque.isEmpty() && deque.peek() < i - k + 1){
                deque.poll();
            }
            // 2.保证单调，每次放进去的数字要比末尾的都大，否则弹出
            while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]){
                deque.pollLast();
            }
            deque.offer(i);
            //因为单调，当i增长到符合第一个k范围的时候，每滑动一步都将队列头节点放入结果
            if(i >= k-1){
                res[idx++] = nums[deque.peek()];
            }
        }
        return res;
    }
}

347. 前 K 个高频元素

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

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int[] result = new int[k];
        HashMap<Integer,Integer> map = new HashMap<>();
        for(int num:nums){
            map.put(num, map.getOrDefault(num,0)+1);
        }
        Set<Map.Entry<Integer,Integer>> entries = map.entrySet();
        PriorityQueue<Map.Entry<Integer,Integer>> queue = new PriorityQueue<>((o1,o2)->o1.getValue()-o2.getValue());
        for(Map.Entry<Integer,Integer> entry : entries){
            queue.offer(entry);
            if(queue.size()>k){
                queue.poll();
            }
        }
        for(int i = k-1 ; i>= 0;i--){
            result[i] = queue.poll().getKey();
        }
        return result;
    }
}