剑指 Offer 59 - I. 滑动窗口的最大值
理解起来不难,可以一直维护最大值及其下标即可,如果窗口内有比当前最大值更大的,则更新最大值即可,如果当前最大值已经不在窗口内了,需要在窗口中重新找过一个最大值。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
if(n == 0) {
return new int[0];
}
int len = n - k + 1;
if (len <= 0) {
return new int[0];
}
int[] res = new int[len];
int l = 0, r = k - 1, idx = 0;
int curMax = Integer.MIN_VALUE, maxIdx = -1;
for (int i = 0; i <= r; i++) {
if (nums[i] > curMax) {
curMax = nums[i];
maxIdx = i;
}
}
while (r < n) {
res[idx] = curMax;
idx++;
r++;
l++;
if (l > maxIdx && r < n) {
curMax = Integer.MIN_VALUE;
for (int i = l; i <= r; i++) {
if (nums[i] > curMax) {
curMax = nums[i];
maxIdx = i;
}
}
}
if (r < n && nums[r] > curMax) {
curMax = nums[r];
maxIdx = r;
}
}
return res;
}
}
这里有一些比较坑的地方在于需要时刻确保窗口没有跑到外面去,要确保窗口右边缘\(r\)没有超过数组范围n。
测试用例也很坑,说好了保证k有效,
还来一个k=0的用例。