解答
首先,大家很容易想到暴力解决的方法。但是我们可以发现,每次移动的时候只有最左边的数会滑出窗口,因此我们可以记录当前最大值的,但是只记录一个值是完全不够的,因为你这个最大值是有可能滑出窗口的。所以我们可以维护一个双端队列来记录有可能成为最大值的位置(为什么是记录位置呢?因为判断出没出窗口是根据位置下标来的),这个队列是按照从大到小的顺序来排列的。
下面我画了图:
具体过程怎么做呢?
假设遍历到了a[i],规则为:
- 如果队列为空,则直接将i放入队列
- 如果队列不为空,则不断取队尾元素与a[i]进行比较,如果比a[i]小,则弹出,再将a[i]放入队列。否则,直接将a[i]放入队列。
- 如果队列头的下标等于i-w(滑出窗口了),则从队列中删除
源代码如下:
import java.util.Collections;
import java.util.ArrayList;
import java.util.LinkedList;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> list = new ArrayList<>();
int len = num.length;
if (num == null || size < 1 || len < size){
return list;
}
LinkedList<Integer> qmax = new LinkedList<>();
for (int i = 0;i < len;i++){
while (!qmax.isEmpty()&&num[qmax.peekLast()] <= num[i]){
qmax.pollLast();
}
qmax.addLast(i);
if (qmax.peekFirst() == i - size){
qmax.pollFirst();
}
if(i >= size - 1)
list.add(num[qmax.peekFirst()]);
}
return list;
}
}