题目:给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。例如,如果输入数组{2, 3, 4, 2, 6, 2, 5, 1}及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为{4, 4, 6, 6, 6, 5},
三种方法:暴力法,最大堆法,双向队列法
//简单方法,时间复杂度O(nW)
int getMax(const int A[], int size)
{
int mx = A[0];
for (int i = 1;i < size;++i)
{
if (A[i] > mx)
mx = A[i];
}
return mx;
}
vector<int> maxInWindows(const int A[], int n, int size)
{
vector<int> result;
if (n < size || size < 0)
return result;
for (int i = 0;i < n - size;++i)
{
int mx = getMax(A + i, size);
result.push_back(mx);
}
return result;
}
//最大堆,时间复杂度O(nlogn)
typedef pair<int, int> Pair;
vector<int> maxInWindows(const vector<int>& nums, int size)
{
vector<int> result;
if (nums.size() < size || size < 1)
return result;
priority_queue<Pair> Q;
for (int i = 0;i < size - 1;++i)
Q.push(Pair(nums[i], i));
for (int i = size - 1;i < nums.size();++i)
{
Q.push(Pair(nums[i], i));
Pair p = Q.top();
while (p.second < i - size + 1)
{
Q.pop();
p = Q.top();
}
result.push_back(p.first);
}
return result;
}
//双向队列
vector<int> maxInWindows(const vector<int>& nums, int size)
{
vector<int> result;
if (nums.size() >= size&&size >= 1)
{
deque<int> index;
for (int i = 0;i < size;++i)
{
while (!index.empty() && nums[i] >= nums[index.back()])
index.pop_back();
index.push_back(i);
}
for (int i = size;i < nums.size();++i)
{
result.push_back(nums[index.front()]);
while (!index.empty() && nums[i] >= nums[index.back()])
index.pop_back();
while (!index.empty() && index.front() <= i - size)
index.pop_front();
index.push_back(i);
}
result.push_back(nums[index.front()]);
}
return result;
}