64. 滑動窗口的最大值

題目描述

給定一個數組和滑動窗口的大小，找出所有滑動窗口裏數值的最大值。例如，如果輸入數組{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]}。

Solution

使用雙端隊列保存滑動窗口內的元素下標，新增值從隊列尾開始比較，去掉隊列尾開始的較小值，這樣能保持隊列頭爲當前窗口的最大值，同時當隊列頭不在窗口內時要移除。

public class Solution {
	    public ArrayList<Integer> maxInWindows(int [] num, int size)
	    {
	        ArrayList<Integer> ret = new ArrayList<>();
	        if (num == null || size < 1 || num.length < size) {
	        	return ret;
	        }
	        LinkedList<Integer> deque = new LinkedList<>();
	        for (int i = 0; i < num.length; i++) {
	            // 去掉隊列尾開始的較小值
	        	while (!deque.isEmpty() && num[i] > num[deque.getLast()]) {
	        		deque.pollLast();
	        	}
	        	deque.offerLast(i);
	        	// 隊列頭不在窗口內時要移除
	        	if (i - deque.getFirst() >= size) {
	        		deque.pollFirst();
	        	}
	        	//數據量已到窗口大小
	        	if (i >= size - 1) {
	        		ret.add(num[deque.peekFirst()]);
	        	}
	        }
	        return ret;
	    }
	}