给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: 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
提示:
你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof
方法一:暴力解决
在获取第一个窗口old的最大值之后,我们移动到下一个窗口,如果原来窗口的最大值old_max还没有被移除,那么就用这个最大值old_max和新添加到窗口的数字arr[ i ]进行比较,从而获得新窗口的最大值,如果old_max更大,那么old_max依旧是新窗口的最大值,否则arr[ i ]是新窗口的最大值,如果原来窗口的最大值已经移除了,那么就需要在这个新窗口的所有数字中重新查找最大值。重复上述操作,直到所有的窗口都已经遍历过了。
对应的代码:
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums == null || nums.length == 0)
return new int[0];
List<Integer> list = new ArrayList<Integer>();
int[] max = getMax(nums,0,k - 1);//获取第一个窗口的最大值以及下标
list.add(max[0]);//将第一个窗口的最大值添加到列表中
int i;
//移动窗口
for(i = 1; i <= nums.length - k; i++){//i表示新窗口的起点
if(max[1] < i){
//如果原来窗口的最大值已经不在了,那么就需要重新查找窗口的最大值
max = getMax(nums,i,i + k - 1);
list.add(max[0]);
}else{
//System.out.println(max[0] + ", " + nums[i]);
if(max[0] <= nums[i + k - 1]){
//如果原来窗口的最大值小于等于新添加到窗口的数(i + k - 1对应的值),那么需要更新最大值,否则不需要更新,直接将其添加到列表中
max[0] = nums[i + k - 1];
max[1] = i + k - 1;
}
list.add(max[0]);
}
}
int[] tar = new int[list.size()];
for(i = 0; i < tar.length; i++){
tar[i] = list.get(i);
}
return tar;
}
//获取数组low-high范围中的最大值以及它的下标
public int[] getMax(int[] nums,int low,int high){
int max = nums[low],max_index = low;
while(low <= high){
if(nums[low] > max){
max = nums[low];
max_index = low;
}
low++;
}
return new int[]{max,max_index};
}
}
运行结果:
方法二:利用队列
根据之前做过的题目:队列的最大值,对应的链接https://leetcode-cn.com/problems/dui-lie-de-zui-da-zhi-lcof/
对应的原理是这样的:
1、获取窗口的最大值,然后将这个最大值添加到队列中
2、如果新添加的数字比队列尾的数字都要大,那么就将队列为尾中的数字跳出,直到队列尾的数字大于等于新添加的数字,此时就将新添加的数字添加到队列尾处。通过这样操作,从而实现了每次移动窗口,只需要判断从窗口中移除的是否是队列中的最大值即可,如果不是,那么就将队列头的数字添加到列表中,否则如果移除的数字是队列中的最大值,那么就将队列头删除,然后队列头的下一个数字就是这个新窗口的最大值,从而不需要再重新查找最大值了。
对应的代码:
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums == null || nums.length == 0)
return new int[0];
List<Integer> list = new ArrayList<Integer>();
Deque<Integer> deque = new LinkedList<Integer>();
int i;
//获取第一个窗口的最大值,然后将最大值添加到队列中去
for(i = 0; i < k; i++)
offer(deque,nums[i]);
list.add(deque.peekFirst());//将第一个窗口的最大值添加到队列中
for(i = 1; i <= nums.length - k; i++){//i是新窗口的起点
if(nums[i - 1] == deque.peekFirst()){
//如果删除的数字是窗口的最大值,那么就需要将最大值从队列中删除
deque.pollFirst();
}
offer(deque,nums[i + k - 1]);//如果删除的数字不是窗口的最大值,那么就将新添加到窗口的数字添加到队列中
list.add(deque.peekFirst());
}
int[] tar = new int[list.size()];
for(i = 0; i < tar.length; i++){
tar[i] = list.get(i);
}
return tar;
}
public void offer(Deque<Integer> deque,int num){
while(!deque.isEmpty() && deque.peekLast() < num){
deque.pollLast();
}
deque.offerLast(num);//将新插入的元素放到队列尾处
}
}
运行结果: