【LeetCode 239】Sliding Window Maximum

Given an array nums, there is a sliding window of size k which is moving from the very left of the array to the very right. You can only see the k numbers in the window. Each time the sliding window moves right by one position.

For example,
Given nums =

[1,3,-1,-3,5,3,6,7]

, and k = 3.

Window position                Max
---------------               -----
[1  3  -1] -3  5  3  6  7       3
1 [3  -1  -3] 5  3  6  7       3
1  3 [-1  -3  5] 3  6  7       5
1  3  -1 [-3  5  3] 6  7       5
1  3  -1  -3 [5  3  6] 7       6
1  3  -1  -3  5 [3  6  7]      7

Therefore, return the max sliding window as

[3,3,5,5,6,7]

.

题意：

　　给定一个数组序列，以及一个固定大小的窗口K，用k从前到后扫描一次数组，返回窗口每个状态下的最大值。

思路：

　　使用双端队列deque当做滑动窗口，保证每个状态下最大值总是在队首。比如[1,3,-1,-3,5,3,6,7]，那么deque的状态分别为[1] 、[3] (前2步，填装滑动窗口) ，（从第三步正式开始）[3,-1]、 [3,-1,-3]、[5]、[5,3]、[6]、[7]，可以看到每次队列的队首元素都是当前滑动窗口的最大值，具体过程可参考代码，有注释。时间复杂度O（n）。

C++:

class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {

vector<int> ret;
if(nums.size() == 0)
return ret;

//双端队列，存储的是元素的下标，不是元素的值
deque<int> slideWindow;
for(int i = 0; i < k; i++)
{
//如果要进队列的数比它前面的数字大，则将它前面的数字从后面删除，直到它前边的数字比它大或者队列为空
while(!slideWindow.empty() && nums[i] >= nums[slideWindow.back()])
{
slideWindow.pop_back();
}

//加入新元素到队列
slideWindow.push_back(i);
}

for(int i = k; i < nums.size(); i++)
{
//队首元素即为当前窗口最大值
ret.push_back(nums[slideWindow.front()]);

//如果要进队列的数比它前面的数字大，则将它前面的数字从后面删除，直到它前边的数字比它大或者队列为空
while(!slideWindow.empty() && nums[i] >= nums[slideWindow.back()])
{
slideWindow.pop_back();
}

//如果当前队首的元素已经不在窗口内部，则将其从队列前边删除
if(!slideWindow.empty() && slideWindow.front() <= i - k)
slideWindow.pop_front();

//加入新元素到队列
slideWindow.push_back(i);
}

//不要忘了最后一个窗口的结果 - -
ret.push_back(nums[slideWindow.front()]);

return ret;
}
};