看到这道题我就有印象, 我在剑指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还高,所以这个算法效率就更高了。