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

 239. 滑动窗口最大值

文章讲解：代码随想录

这道题我写了两种解法，第一种是双端队列的思想，第二种是优先队列的思想；

双端队列的思想：每次元素从队尾入队列，进入队列之前判断一下队尾的元素和当前元素对比，如果当前元素大于队尾所对应的元素，那么将队尾元素取出，如果队尾元素不大于当前元素，将当前元素入队（队列存储的是数组的下标）；检查队头和当前元素下标位置只差是否大于等于k，如果是的话，说明队头对应的下标元素不在滑动窗口中，需要移除，最后就是将队头的元素添加到结果中。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int[] res = new int[nums.length - k + 1];
        Deque<Integer> deque = new LinkedList<>();
        int index = 0;
        while (index < k) {
            while(!deque.isEmpty()&&nums[index]>nums[deque.peekLast()]){
                deque.pollLast();
            }
            deque.offer(index);
            index++;
        }
        res[0] = nums[deque.peekFirst()];
        int bj = 1;
        for (int i = index; i < nums.length; i++) {
            while(!deque.isEmpty()&&nums[i]>nums[deque.peekLast()]){
                deque.pollLast();
            }
          while(!deque.isEmpty()&&(i-k)>=deque.peekFirst()){
                deque.pollFirst();
            }
            deque.offer(i);
            res[bj++] =  nums[deque.peekFirst()];
        }
        return res;
    }
}

优先队列的思想：优先队列就是将该数组的下标对应的元素大小依次加入到队列中，同样队列中存储的是数组的下标，每次添加元素前，先判断优先队列中队头中的元素是否在滑动窗口中，如果不在的话，将该元素出队，循环此操作，然后元素入队，获取队头元素添加到结果中。
 

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int[] res = new int[nums.length - k + 1];
        Deque<Integer> deque = new LinkedList<>();
        int index = 0;
        while (index < k) {
            while(!deque.isEmpty()&&nums[index]>nums[deque.peekLast()]){
                deque.pollLast();
            }
            deque.offer(index);
            index++;
        }
        res[0] = nums[deque.peekFirst()];
        int bj = 1;
        for (int i = index; i < nums.length; i++) {
            while(!deque.isEmpty()&&nums[i]>nums[deque.peekLast()]){
                deque.pollLast();
            }
          while(!deque.isEmpty()&&(i-k)>=deque.peekFirst()){
                deque.pollFirst();
            }
            deque.offer(i);
            res[bj++] =  nums[deque.peekFirst()];
        }
        return res;
    }
}

---

 347.前 K 个高频元素

文章讲解：代码随想录

思路：首先创建一个HashMap 集合，然后key 存储的元素，value 存储元素的个数。然后我们创建一个优先队列，排序规则是根据key 所对应的value 值进行排序，然后在优先队列中取出k 个数即可。

 public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            hashMap.put(nums[i], hashMap.getOrDefault(nums[i], 0) + 1);
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> Integer.compare(hashMap.get(o2), hashMap.get(o1)));
        Iterator<Integer> iterator = hashMap.keySet().iterator();
        while (iterator.hasNext()) {
            queue.offer(iterator.next());
        }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = queue.poll();
        }
        return res;
    }

总时长：2个小时30分钟