代码随想录算法训练营Day13|栈与队列part03

代码随想录算法训练营Day13|栈与队列part03

---

---

一、239. 滑动窗口最大值(好难)

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        //使用双端队列进行存储窗口的所有的值
        //如果存入较大的值，就要弹出前面所有比他小的值
        //小的就直接入队列
        //并且保证队列里面十窗口的合法值
        int n=nums.length;
        int re[]=new int[n-k+1];
        ArrayDeque<Integer> ad=new ArrayDeque<>();
        int index=0;
        for(int i=0;i<n;i++){
 // 1.队列头结点需要在[i - k + 1, i]范围内，不符合则要弹出
            while(!ad.isEmpty()&&ad.peek()<(i-k+1)){
                ad.poll();
            }

  // 2.既然是单调，就要保证每次放进去的数字要比末尾的都大，否则也弹出
       while(!ad.isEmpty() && nums[ad.peekLast()] < nums[i]) {
                ad.pollLast();
            }
        
        ad.offer(i);
 // 因为单调，当i增长到符合第一个k范围的时候，每滑动一步都将队列头节点放入结果就行了
            if(i >= k - 1){
                re[index++] = nums[ad.peek()];
            }
        }
        return re;
    }
}

二、347. 前 K 个高频元素（难）

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++){//依次从队头弹出k个,就是出现频率前k高的元素
            ans[i] = pq.poll()[0];
        }
        return ans;
    }
    }