题目
有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7],w=3为例。因为第一个窗口[4,3,5]的最大值为5,第二个窗口[3,5,4]的最大值为5,第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。
给定整形数组arr及它的大小n,同时给定w,请返回res数组。保证w小于等于n,同时保证数组大小小于等于500。
测试样例:[4,3,5,4,3,3,6,7],8,3
返回:[5,5,5,4,6,7]
思路
用双端队列deque保存可能成为滑动窗口中最大元素的下标,进出队列规则如下:
- 如果deque为空,直接将下标i放入deque中。
- 如果当前下标i-队头元素==w,弹出队头元素。
- 如果deque不为空,取出deque队尾的下标j,如果arr[j]>=arr[i],将下标i放入deque队尾。否则依次弹出队尾元素,直到arr[j]<arr[i]或者deque为空,再将i压入队尾。
1和2都比较好理解,这里解释一下3:
如果arr[j]>=arr[i],将下标i放入deque队尾。否则依次弹出队尾元素,直到arr[j]<arr[i]或者deque为空,再将i压入队尾.
arr[j]>=arr[i],为什么要放在队尾呢,arr[i]虽然此时比arr[j]小,但是arr[j]会在一定时间后失效(即脱出滑动窗口范围),arr[i]可能会成为最大值。
而当arr[j]<arr[i],说明arr[j]永远不能成为当前窗口及以后窗口的最大值了,因为arr[j]不仅比arr[i]小,而且一定比arr[j]先失效。所以将其弹出即可。
当初始队列构造完成后,依次读入数组元素,队列的头元素的下标所代表的数字就是此时窗口中的最大值。
原文:https://www.jianshu.com/p/3242ddf81428
原创部分:
原文的解释是很到位的,我再补充一点儿,算法复杂度为O(n),可能乍一看,会觉得这个复杂度是O(nw)因为规则3有一个向前遍历的过程。其实不然,我们可以这样考虑,对于数组中的每个元素来说,最多入队,出队一次,所以这个时间复杂度就是O(n)
以下是我给出的java代码实现:
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
if(size <= 0)
return new ArrayList<Integer>();
Deque<Integer> dq = new LinkedList<>();
ArrayList<Integer> al = new ArrayList<>();
for(int i=0; i<num.length; i++){
if(dq.isEmpty()) //规则1
dq.addFirst(i);
else{ //规则3
while(!dq.isEmpty()){
if(num[dq.getLast()] < num[i])
dq.removeLast();
else break;
}
dq.addLast(i);
}
if(i-dq.getFirst() == size) //规则2
dq.removeFirst();
if(i>=size-1) //当凑够了一个窗口才开始添加最大值
al.add(num[dq.getFirst()]);
}
return al;
}
从这里我们也可以看出,代码固然重要,但是更重要的是思想。不过也要有扎实的基本功,否则你连Deque怎么用都不知道,那就没办法把思路转换成代码,用代码实现思路这个过程需要大量的积累和练习。