给你一个整数数组 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 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
1 <= k <= nums.length
这道题需要我们求解滑动窗口内的最大值。滑动窗口是固定大小为k的子数组,从数组左端开始,每次向右移动一个位置,直到窗口右端达到数组末尾。
关键理解:
n-k+1
的数组,其中n是原数组长度例如,对于示例1中的数组[1,3,-1,-3,5,3,6,7]
和k=3:
[1,3,-1]
,最大值是3[3,-1,-3]
,最大值是3[-1,-3,5]
,最大值是5最终输出:[3,3,5,5,6,7]
最直接的思路是对每个窗口位置,遍历窗口内的所有元素来找出最大值。
具体步骤:
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
if (n == 0 || k == 0) {
return new int[0];
}
// 结果数组长度为 n-k+1
int[] result = new int[n - k + 1];
// 对每个窗口位置
for (int i = 0; i <= n - k; i++) {
int max = nums[i]; // 初始化为窗口内第一个元素
// 遍历窗口内的其他元素
for (int j = i + 1; j < i + k; j++) {
max = Math.max(max, nums[j]);
}
result[i] = max; // 保存当前窗口的最大值
}
return result;
}
}
n-k+1
暴力法适用于数组长度不大且窗口大小较小的情况。当数组长度或窗口大小较大时,这种方法效率较低。
我们可以使用优先队列(最大堆)来维护窗口中的元素,这样可以在O(log k)时间内获取窗口中的最大值。
为了处理元素移出窗口的情况,我们需要在队列中存储元素值和索引的对,这样可以判断堆顶元素是否还在当前窗口内。
具体步骤:
import java.util.PriorityQueue;
import java.util.Comparator;
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
if (n == 0 || k == 0) {
return new int[0];
}
// 结果数组
int[] result = new int[n - k + 1];
// 创建最大堆,按元素值降序排列
PriorityQueue<int[]> maxHeap = new PriorityQueue<>(
(a, b) -> b[0] - a[0] // 降序排列,最大值在堆顶
);
// 初始将前k个元素加入堆
for (int i = 0; i < k; i++) {
maxHeap.offer(new int[]{
nums[i], i});
}
// 获取第一个窗口的最大值
result[0] = maxHeap.peek()[0];
// 处理后续窗口
for (int i = k; i < n; i++) {
// 添加当前元素到堆
maxHeap.offer(new int[]{
nums[i], i});
// 移除不在当前窗口内的元素
while (!maxHeap.isEmpty() && maxHeap.peek()[1] <= i - k) {
maxHeap.poll();
}
// 当前窗口的最大值
result[i - k + 1] = maxHeap.peek()[0];
}
return result;
}
}
n-k+1
优先队列法适用于窗口大小较大的情况,因为它对每个窗口的处理不需要遍历所有k个元素。但当k很小时,暴力法可能更快。
双端队列法是本题的最优解法之一。它通过维护一个单调递减的队列(存储元素索引),保证队首始终是当前窗口的最大元素索引。
关键思想:
具体步骤:
import java.util.ArrayDeque;
import java.util.Deque;
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
if (n == 0 || k == 0) {