滑动窗口最大值
题目:
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
示例 3:
输入:nums = [1,-1], k = 1
输出:[1,-1]
示例 4:
输入:nums = [9,11], k = 2
输出:[11]
示例 5:
输入:nums = [4,-2], k = 2
输出:[4]
解题思路1:用优先队列进行解决
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int[] ans = new int[nums.length - k + 1];
int cur = 0;
PriorityQueue<int[]> queue = new PriorityQueue<int[]>((a, b) -> {
if(a[0] == b[0]) {
return b[1] - a[1];
}
return b[0] - a[0];
});
for(int i = 0; i < k; i++) {
queue.offer(new int[]{nums[i], i});
}
ans[0] = queue.peek()[0];
for(int i = k; i < nums.length; i++) {
queue.offer(new int[]{nums[i], i});
while(queue.peek()[1] <= i - k) {
queue.poll();
}
ans[i - k + 1] = queue.peek()[0];
}
return ans;
}
}
解题思路2:用双向队列解决
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int[] ans = new int[nums.length - k + 1];
Deque<Integer> queue = new LinkedList();
for(int i = 0; i < k; i++) {
while(!queue.isEmpty() && nums[i] >= nums[queue.peekLast()]) {
queue.pollLast();
}
queue.offerLast(i);
}
ans[0] = nums[queue.peekFirst()];
for(int i = k; i < nums.length; i++) {
while(!queue.isEmpty() && nums[i] >= nums[queue.peekLast()]) {
queue.pollLast();
}
queue.offerLast(i);
while(queue.peekFirst() <= i - k) {
queue.pollFirst();
}
ans[i - k + 1] = nums[queue.peekFirst()];
}
return ans;
}
}