题目描述:给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组 {2,3,4,2,6,2,5,1} 及滑动窗口的大小3,那么一共存在6个滑动窗口:{[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]},他们的最大值分别为{4,4,6,6,6,5}。
解决思路:
- 暴力方法。在每个窗口内都找到最大值,如果数组为n,窗口为m,那么时间复杂度为O(mn)。
import java.util.*;
public class Main {
public static void main(String[] args) {
int[] array = new int[]{2, 3, 4, 2, 6, 2, 5, 1};
int slide = 3;
ArrayList<Integer> list = new ArrayList<>();
for(int i = 0; i < array.length - 2; i++){
int max = array[i];
for(int j = i; j < i + 3; j++){
if(array[j] > max)
max = array[j];
}
list.add(max);
}
for (Integer e : list) {
System.out.println(e);
}
}
}
- 第二种解法是利用单调队列。从头到尾遍历数组时,根据如下的规则出队入队:
1)如果队列为空,则当前队列入队。
2)如果当前数字大于等于队列尾,则删除队列尾,知道当前数字小于队列尾部。
3)如果当前数字小于队列尾,则当前数字入队列
4)如果队列头超出滑动窗口范围,则删除队列头。
import java.util.*;
public class Main {
public ArrayList<Integer> slideWindow(int[] array, int w){
if(array == null || w < 1 || array.length < w)
return null;
ArrayList<Integer> result = new ArrayList<>();
LinkedList<Integer> queue = new LinkedList<>();//只保存数组的下标,便于判断比较与滑动窗口的大小
for(int i = 0; i < array.length; i++){
while (!queue.isEmpty() && array[i] >= array[queue.peekLast()]){
queue.pollLast();
}
queue.addLast(i);
//队列头超过滑动窗口的范围
if(i - w == queue.peekFirst())
queue.pollLast();
if(i >= w - 1)
result.add(array[queue.peekFirst()]);
}
return result;
}
public static void main(String[] args) {
int[] array = new int[]{3, 3, 2, 2};
int w = 3;
ArrayList<Integer> res = new Main().slideWindow(array, w);
for (Integer e:res) {
System.out.println(e);
}
}
}