要求
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
思路
元素是从窗口的右侧进入的,然后由于窗口大小是固定的,因此多余的元素是从窗口左侧移除的。 一端进入,另一端移除,这不就是队列的性质吗?所以,该题目可以借助队列来求解。
当考察到第五个元素5时,因此需要剔除当前队列末尾的数字-3,但队首有元素3存在,因此不能将其从队首移除,只能将其从队尾移除,直到,队列为空或当前考察元素小于新的队尾元素;一端既可以有元素入队,又有元素出队的队列,称之为双向队列。并且当队首元素在原数组中的下标小于窗口左边界时,队首元素就需要移除队列。
解析要求:
- 遍历给定数组中的元素,如果队列不为空且当前考察元素大于等于队尾元素,则将队尾元素移除。直到(这里是重点,记得用循环,而不是if判断),队列为空或当前考察元素小于新的队尾元素;
- 当队首元素的下标小于滑动窗口左侧边界left时,表示队首元素已经不再滑动窗口内,因此将其从队首移除。
- 由于数组下标从0开始,因此当窗口右边界right+1大于等于窗口大小k时,意味着窗口形成。此时,队首元素就是该窗口内的最大值。
public class LeetCode239 {
public int[] maxSlidingWindow(int[] nums, int k) {
//存放最大值的个数
int[] res = new int[nums.length - k + 1];
LinkedList<Integer> queue = new LinkedList<>();
//遍历数组中元素,right代表滑动窗口的右边界
for (int right = 0; right < nums.length; right++) {
//如果队列不为空且当前考察的元素大于等于队尾元素,则将队尾元素移除;直到队列元素为空或者新得队尾元素大于考察元素
while (!queue.isEmpty() && (nums[right] >= nums[queue.peekLast()])){
queue.removeLast();
}
//存储元素的下标
queue.addLast(right);
//计算左窗口的边界,当队首元素的下标小于窗口的左边界时left时,则将队首元素移除
int left = right - k + 1;
if (queue.peekFirst() < left){
queue.removeFirst();
}
//由于数组下标从0开始,当窗口右边界right + 1大于等于窗口大小k时,此时队首元素就是该窗口内的最大值
if (right + 1 >= k){
res[left] = nums[queue.peekFirst()];
}
}
return res;
}
}