给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
我的思路:采用双端队列,通过双重循环遍历,每k个一组,依次加入双端队列,同时调用一个函数,将该组的最大值移到最左边。该轮次结束,将双端队列清空,进行下一轮。
写法一:双端队列+双重循环
// 双端队列
static ArrayDeque<Integer> deque = new ArrayDeque<>();
static public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 0) {
return new int[0];
}
if (k == 1) {
return nums;
}
List<Integer> list = new ArrayList<Integer>();
for (int i = 0; i < nums.length - k + 1; i++) {
for (int j = i; j < i + k; j++) {
deque.add(nums[j]);
maxNum();
}
list.add(deque.getFirst());
deque.clear();
}
int[] arr = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
arr[i] = list.get(i);
}
return arr;
}
static public void maxNum() {
while (deque.getFirst() < deque.getLast()) {
deque.removeFirst();
}
}
}
采用双端队列,是因为之前看过某个视频,那种做法我没有吸收好,然后自己就捣鼓了一下,但是性能很差,耗时长。这种做法其实不用双端队列都可以,下面的写法相同的思想:
写法二:双重循环
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length * k == 0) {
return new int[0];
}
if (nums.length * k == 1) {
return nums;
}
int[] res = new int[nums.length - k + 1];
for (int i = 0; i < nums.length; i++) {
int max = Integer.MIN_VALUE;
for (int j = i; j < i + k; j++) {
if (nums[j] > max) {
max = nums[j];
}
}
res[i] = max;
}
return res;
}
详讲:这里的res数组的长度,根据nums.length- k +1 可以获得。每k个元素一轮,找出最大的值,加入res数组。
还有一种高效的双端队列做法,但是现阶段,修炼不够,日后再去理解理解,待续。