LeetCode239.滑动窗口最大值

LeetCode239.滑动窗口最大值_第1张图片

看到这道题我就有印象, 我在剑指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 queue = new PriorityQueue<>(new Comparator<>(){
           public int compare(Integer o1, Integer o2){
               return o2-o1;
           }
       });
       for(int i=0;i

LeetCode239.滑动窗口最大值_第2张图片

我是没想到我用一遍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 queue = new PriorityQueue<>(new Comparator(){
           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

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

你可能感兴趣的:(leetcode,算法,leetcode,数据结构,java)