题目一:滑动窗口的最大值
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为{4,4,6,6,6,5}。
解题思路
实际上,一个滑动窗口可以看成一个队列。当滑动窗口时,处于窗口的第一个数字被删除,同时在窗口的末尾添加一个新的数字。这符合队列的“先进先出”特性。如果能从队列中找出它的最大数,那么这个问题就解决了。
我们并不把滑动窗口的每个数值都存入队列,而是只把有可能成为滑动窗口最大值的数值存入一个两端开口的队列,接着输入数组{2,3,4,2,6,2,5,1}进行分析。如下表
C++实现
class Solution
{
public:
vector<int> maxInWindow(const vector<int>&nums,unsigned int size)
{
int n = nums.size();
deque<int> dqu;
vector<int>maxInWindows;
if (!n||size>n||size<1)
return vector<int>();
for (unsigned int i = 0; i < size; ++i)
{
while (!dqu.empty() && nums[i] >= nums[dqu.back()])
dqu.pop_back();
dqu.push_back(i);
}
for (unsigned int i = size; i < n; ++i)
{
maxInWindows.push_back(nums[dqu.front()]);
while (!dqu.empty() && nums[i] >= nums[dqu.back()])
dqu.pop_back();
if (!dqu.empty() && dqu.front() <= (int)(i - size))
dqu.pop_front();
dqu.push_back(i);
}
maxInWindows.push_back(nums[dqu.front]);
return maxInWindows;
}
};
- 时间复杂度:O(n),数组单调递增的时候复杂度最差,为O(n*size),且size为常数,所以时间复杂度为O(n)
- 空间复杂度:O(n)
题目二:队列的最大值
请定义一个队列并实现函数max得到队列里的最大值,要求函数max、push_back和pop_front的时间复杂度都是O(1)。
解题思路
如前所述,滑动窗口可以看成一个队列,因此上题的解法可以用来实现带max函数的队列。
C++实现
#pragma once
#include<deque>
#include <exception>
template<typename T>class QueueWithMax
{
public:
QueueWithMax():currentIndex(0) {}
void push_back(T number);
void pop_front();
T max()const;
private:
struct InternalData
{
T number;
int index;
};
std::deque <InternalData>data;
std::deque <InternalData>maxnums;
int currentIndex;
};
template<typename T>void QueueWithMax<T>::push_back(T number)
{
while (!maxnums.empty() && number >= maxnums.back().number)
maxnums.pop_back();
InternalData internalData = { number,currentIndex };
data.push_back(internalData);
maxnums.push_back(internalData);
}
template<typename T>void QueueWithMax<T>::pop_front()
{
if (maxnums.empty())
throw new exception("query is empty");
if (data.front().index == maxnums.front().index)
maxnums.pop_front();
data.pop_front();
}
template<typename T>T QueueWithMax<T>::max()const
{
if(maxnums.empty())
throw new exception("query is empty");
return maxnums.front().number;
}