leetcode 239. 滑动窗口最大值
问题描述
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
示例:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值
--------------- -----
[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
提示:
你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。
进阶:
你能在线性时间复杂度内解决此题吗?
代码1(优先队列)
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
priority_queue<pair<int, int>> q;
vector<int> ans;
int n = nums.size(),i;
for(i = 0; i < n; i++)
{
//队列q中的元素达到k个了,需要移走最右边队头的元素
while(!q.empty() && q.top().second <= i-k)
q.pop();
//把新元素加入队列,优先队列自动排序
q.push({nums[i],i});
//当i >= k-1时,可以产生答案了,队头存储了最大元素的下标
if(i >= k-1)ans.push_back(q.top().first);
}
return ans;
}
};
结果:
执行用时 :44 ms, 在所有 C++ 提交中击败了34.75%的用户
内存消耗 :14.7 MB, 在所有 C++ 提交中击败了11.83%的用户
代码2(双向队列)
注意pair的顺序,第一个是数值,第二个是对应下标
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> dq;
vector<int> ans;
int n = nums.size(),i;
for(i = 0; i < n; i++)
{
//双向队列dq中的元素达到k个了,需要移走最右边队头的元素
if(!dq.empty() && dq.front() == i-k)
dq.pop_front();
//从队尾开始比较与新元素的大小,如果队尾的元素比新元素小则从队列中去掉,即队列是从高到底排的
while(!dq.empty() && nums[dq.back()] < nums[i])
dq.pop_back();
//把新元素假如队列
dq.push_back(i);
//当i >= k-1时,可以产生答案了,队头存储了最大元素的下标
if(i >= k-1)ans.push_back(nums[dq.front()]);
}
return ans;
}
};
结果:
执行用时 :24 ms, 在所有 C++ 提交中击败了73.25%的用户
内存消耗 :13.3 MB, 在所有 C++ 提交中击败了25.27%的用户