题目描述:
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
方法一:最大堆(优先队列),priority_queue
对于本题而言,初始时,我们将数组 \textit{nums}nums 的前 kk 个元素放入优先队列中。每当我们向右移动窗口时,我们就可以把一个新的元素放入优先队列中,此时堆顶的元素就是堆中所有元素的最大值。然而这个最大值可能并不在滑动窗口中,在这种情况下,这个值在数组nums中的位置出现在滑动窗口左边界的左侧。因此,当我们后续继续向右移动窗口时,这个值就永远不可能出现在滑动窗口中了,我们可以将其永久地从优先队列中移除。
我们不断地移除堆顶的元素,直到其确实出现在滑动窗口中。此时,堆顶元素就是滑动窗口中的最大值。为了方便判断堆顶元素与滑动窗口的位置关系,我们可以在优先队列中存储二元组
我们可以采用最大堆的数据结构来保存元素,堆顶元素即为当前堆的最大值,并判断当前堆顶元素这是否在窗口中,在则直接返回,不在则删除堆顶元素并调整堆。
我们以nums = [1,3,-1,-3,5,3,6,7], k = 3为例来模拟堆的过程。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int n = nums.size();
priority_queue<pair<int, int>> q; //优先队列
for (int i = 0; i < k; ++i) { //先把前k个元素加入到优先队列中
q.emplace(nums[i], i);
}
vector<int> ans
ans.push_back(q.top().first);
for (int i = k; i < n; ++i) {
q.emplace(nums[i], i);
while (q.top().second <= i - k) { //i-k表示窗口的最左边界
q.pop();
}
ans.push_back(q.top().first);
}
return ans;
}
};
方法二:单调队列(双端队列)
首先,明确一个概念,单调队列是在队列中所有的元素都是单调的,要么单调增,要么单调减,新增元素时到队尾,此时队列不符合单调性就将队尾元素出队,直到队列单调或空。
还是以nums = [1,3,-1,-3,5,3,6,7], k = 3为例来模拟对列的过程。(队列中的元素是单调递减的)
作者:capital-worker
链接:https://leetcode.cn/problems/sliding-window-maximum/solution/you-xian-dui-lie-zui-da-dui-dan-diao-dui-dbn9/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。