2、数组和滑动窗口
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下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]}。
思路一:利用队列
使用队列,遍历整个数组,三个三个元素这样逐次添加到对列中;
每当从对列中取出一个元素之后,都要向队列中添加一个元素
package D栈;
import com.sun.jmx.remote.internal.ArrayQueue;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Queue;
/**
* @Author Zhou jian
* @Date 2020 ${month} 2020/2/15 0015 15:46
* 给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下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]}。
*/
public class Problem2 {
public static void main(String[] args) {
Problem2 problem2 = new Problem2();
int[] num ={2,3,4,2,6,2,5,1};
System.out.println(problem2.maxInWindows(num,3));
}
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> arrayList = new ArrayList<>();
//借助队列
Queue<Integer> queue = new ArrayDeque<>();
//队列中首先添加size个元素
for(int i=0;i<size;i++){
queue.add(num[i]);
}
for(int t=size;t<num.length;t++){
int max =max(queue);//获取目前队列中3个元素的最大值
arrayList.add(max);//将队列元素中的最大值加入到ArrayList
queue.poll();//取出一个元素
queue.add(num[t]);//介入一个元素
}
return arrayList;
}
public int max(Queue<Integer> queue){
int max = queue.element();
Object[] arr = queue.toArray();
for(int i=0;i<arr.length;i++){
int value = (int)arr[i];
if(value>max){
max = value;
}
}
return max;
}
}
思路二:利用队列改进
并不把滑动窗口的每个数值都存入队列,而是只把有可能成为滑动窗口最大值的数值存入队列。
private static List<Integer> maxInWindows(List<Integer> data, int size) {
List<Integer> windowMax = new LinkedList<>();
// 条件检查
if (data == null || size < 1 || data.size() < 1) {
return windowMax;
}
Deque<Integer> idx = new LinkedList<>();
// 窗口还没有被填满时,找最大值的索引
for (int i = 0; i < size && i < data.size(); i++) {
// 如果索引对应的值比之前存储的索引值对应的值大或者相等,就删除之前存储的值
while (!idx.isEmpty() && data.get(i) >= data.get(idx.getLast())) {
idx.removeLast();
}
// 添加索引
idx.addLast(i);
}
// 窗口已经被填满了
for (int i = size; i < data.size(); i++) {
// 第一个窗口的最大值保存
windowMax.add(data.get(idx.getFirst()));
// 如果索引对应的值比之前存储的索引值对应的值大或者相等,就删除之前存储的值
while (!idx.isEmpty() && data.get(i) >= data.get(idx.getLast())) {
idx.removeLast();
}
// 删除已经滑出窗口的数据对应的下标
if (!idx.isEmpty() && idx.getFirst() <= (i - size)) {
idx.removeFirst();
}
// 可能的最大的下标索引入队
idx.addLast(i);
}
// 最后一个窗口最大值入队
windowMax.add(data.get(idx.getFirst()));
return windowMax;
}