剑指 Offer 59 - I. 滑动窗口的最大值(复习)
题目:
给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
解答:
转

class Solution { public int[] maxSlidingWindow(int[] nums, int k) { if(nums.length<k || k<=0) return new int[0]; Deque<Integer> deque = new LinkedList<>(); int[] res = new int[nums.length-k+1]; for(int i = 0;i<k;i++){ while(!deque.isEmpty() && deque.peekLast()<nums[i]) deque.removeLast(); deque.addLast(nums[i]); } res[0] = deque.peekFirst(); for(int j = k;j<nums.length;j++){ if(deque.peekFirst() == nums[j-k]) deque.removeFirst(); while(!deque.isEmpty() && deque.peekLast()<nums[j]) deque.removeLast(); deque.addLast(nums[j]); res[j-k+1] = deque.peekFirst(); } return res; } }
方法二:大根堆 + 双指针
class Solution { public int[] maxSlidingWindow(int[] nums, int k) { if(nums.length<k || k<=0 ||nums.length<=0||nums == null) return new int[0]; PriorityQueue<Integer> heap = new PriorityQueue<>((o1,o2)->o2-o1); int l = 0, r = k-1,index = 0,cnt = 0; int[] res = new int[nums.length-k+1]; for(int j = 0;j<k;j++) heap.offer(nums[j]); for(;r<nums.length;){ res[index] = heap.peek(); index++; heap.remove(nums[l]); l++; r++; if(r<nums.length) heap.add(nums[r]); } // for(;cnt<k;cnt++) // heap.offer(nums[cnt]); // while(cnt <nums.length){ // res[index] = heap.peek(); // index++; // heap.remove(nums[cnt-k]); // heap.add(nums[cnt]); // cnt++; // } //res[index] = heap.peek(); return res; } }

浙公网安备 33010602011771号