文章目录
- 一、题目
- 二、解法
- 三、完整代码
所有的LeetCode题解索引,可以看这篇文章——【算法和数据结构】LeetCode题解。
一、题目
二、解法
思路分析:这道题我们如果用暴力破解法需要 O ( n ∗ k ) O(n*k) O(n∗k)的复杂度。思索再三,我们需要一个能够把最大值放在队头,整个队列单调递减的单调队列。每次窗口移动的时候,调用que.pop(滑动窗口中移除元素的数值),que.push(滑动窗口添加元素的数值),然后que.front()就返回我们要的最大值。要注意两点:1、最大值必须在队头,这样我们才能用que.front()访问 2、队头的最大值元素必须在滑动窗口内部,否则就不是滑动窗口的最大值。
对于上面两个问题我们分别在pop和push中做这样的逻辑:1、如果队尾元素如果小于入队元素,就弹出队尾元素,直到入队元素大于等于队尾元素才插入入队元素 2、每次访问que.front()之前,都进行一次判断,如果队头的最大值和nums[i-k]进行对比,如果相等,说明这个最大值元素不在滑动窗口中(滑动窗口范围为i-k+1,…,i-1,i)。
程序如下:
class MyQueue { // 单调队列,递减
public:
deque<int> que; // deque为双向数组,这里用它当做单调队列的底层实现
void pop(int value) {
if (!que.empty() && value == que.front()) // 如果元素相等则弹出,否则不做操作
que.pop_front();
}
void push(int value) { // 保证队头元素一定是最大的
while (!que.empty() && value > que.back()) { // 队尾元素如果小于入队元素,就弹出队尾元素
que.pop_back();
}
que.push_back(value); // 直到入队元素大于等于队尾元素,插入
}
int front() {
return que.front();
}
};
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que; // 单调队列
vector<int> result;
for (int i = 0; i < k; ++i) {
que.push(nums[i]);
}
result.push_back(que.front());
for (int i = k; i < nums.size(); ++i) {
que.pop(nums[i - k]); // 弹出队头的元素,防止最大值一直在队头,且这个最大值又不在滑动窗口中的情况
que.push(nums[i]);
result.push_back(que.front());
}
return result;
}
};
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n),所有元素的push和pop操作都只进行一次。
- 空间复杂度: O ( k ) O(k) O(k),需要一个大小为k的单调队列额外空间。
三、完整代码
# include <iostream>
# include <vector>
# include <deque>
using namespace std;
class MyQueue { // 单调队列,递减
public:
deque<int> que; // deque为双向数组,这里用它当做单调队列的底层实现
void pop(int value) {
if (!que.empty() && value == que.front()) // 如果元素相等则弹出,否则不做操作
que.pop_front();
}
void push(int value) { // 保证队头元素一定是最大的
while (!que.empty() && value > que.back()) { // 队尾元素如果小于入队元素,就弹出队尾元素
que.pop_back();
}
que.push_back(value); // 直到入队元素大于等于队尾元素,插入
}
int front() {
return que.front();
}
};
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que; // 单调队列
vector<int> result;
for (int i = 0; i < k; ++i) {
que.push(nums[i]);
}
result.push_back(que.front());
for (int i = k; i < nums.size(); ++i) {
que.pop(nums[i - k]); // 弹出队头的元素,防止最大值一直在队头,且这个最大值又不在滑动窗口中的情况
que.push(nums[i]);
result.push_back(que.front());
}
return result;
}
};
void my_print(vector <int>& v, string msg)
{
cout << msg << endl;
for (vector<int>::iterator it = v.begin(); it != v.end(); it++) {
cout << *it << " ";
}
cout << endl;
}
void VectorGenerator(int* arr, vector<int>& v, int arr_len) {
for (int i = 0; i < arr_len; ++i) {
v.push_back(arr[i]);
}
}
int main()
{
int window_size = 3;
int arr[] = {1, 3, -1, -3, 5, 3, 6, 7};
int arr_len = sizeof(arr) / sizeof(int);
vector<int> nums;
VectorGenerator(arr, nums, arr_len);
my_print(nums, "目标数组");
Solution s1;
vector<int> result = s1.maxSlidingWindow(nums, window_size);
my_print(result, "滑动窗口最大值");
system("pause");
return 0;
}
end