LeetCode239.滑动窗口最大值

看到这道题我就有印象， 我在剑指offer里面做过这道题，我记得当时用的是优先队列，然后我脑子里一下子就有了想法，拿优先队列作为窗口，每往右移动一步，把左边的数remove掉，把右边的数add进来，然后把队头，也就是窗口中最大的元素放入答案数组，然后就写出了如下代码：

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
       int n = nums.length;
       int[] ans = new int [n-k+1];
       int index =0;
       PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<>(){
           public int compare(Integer o1, Integer o2){
               return o2-o1;
           }
       });
       for(int i=0;i<k;i++){
           queue.add(nums[i]);
       }
       ans[index++] = queue.peek();
       for(int i=k;i<n;i++){
           queue.remove(nums[i-k]);
           queue.add(nums[i]);
           ans[index++] = queue.peek();
       }
       return ans;
    }
}

我是没想到我用一遍for循环都能超时，然后我就想了下觉得没问题啊，我之前是这样做的啊。然后看了一下自己之前写的题解:

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
       int n = nums.length;
       int[] ans = new int [n-k+1];
       int index =0;
       PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>(){
           public int compare(int[] o1, int[] o2){
               return o2[0]!=o1[0] ? o2[0]-o1[0] : o2[1]-o1[1];
           }
       });
       for(int i=0;i<k;i++){
           queue.add(new int[]{nums[i], i});
       }
       ans[index++] = queue.peek()[0];
       for(int i=k;i<n;i++){
           queue.add(new int[]{nums[i], i});
           while(queue.peek()[1] <= i-k){
               queue.poll();
           }
           ans[index++] = queue.peek()[0];
       }
       return ans;
    }
}

队列里面存的不是数组的元素，而是一个大小为2的数组(其实有点像k-v键值对)，数组第0个元素是参数数组nums中的值，第1个元素是这个值在nums中的下标，这样的话他就少了remove的一步，我是add一个就remove一个，这个是通过下标判断，如果队头这个最大的元素的下标在窗口的左边就直接poll，否则不用管，这样既少了每次remove而且poll的效率比remove还高，所以这个算法效率就更高了。