题目
给你一个整数数组?nums
,有一个大小为?k
?的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的?k
?个数字。滑动窗口每次只向右移动一位。
返回?滑动窗口中的最大值?。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1 输出:[1]
提示:
1 <= nums.length <= 105
-104?<= nums[i] <= 104
1 <= k <= nums.length
滑动窗口
这道题和之前的滑动窗口的题目有点类似,要避免直接两层循环暴力求解,可以使用滑动窗口【LeetCode热题100】【滑动窗口】找到字符串中所有字母异位词_找到字符串中所有字母异位 题解-CSDN博客
要寻找这个滑动窗口的最大值最快的方法是使用一个大顶堆,堆的插入元素的时间复杂度为logn,这样不用遍历窗口的每个元素就可以找出最大值
但这样还有一个问题,那就是滑动窗口移动的时候,如果删除左边被移出窗口的元素,堆删除指定元素并不简单,解决的方法就是不删除,当堆顶元素为已经移出窗口的元素时,pop堆顶元素就行,这样就可以避免找到的最大值是已经移除的元素
为了实现判断这个元素是否已经移除窗口,我们采用二元组来存储每个元素本身和它的索引,当索引小于等于当前的i-k,说明这个元素已经不在了,做掉
还有一点就是C++没有堆这个容器,但是有优先队列,这个是堆实现的可以当成堆来用,而且默认是大顶堆,而C++中的二元组pari比较大小的时候默认是先比较第一个,所以我们需要把元素本身放第一个
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int>answer;
priority_queue<pair<int,int>>deap;
for(int i=0;i<k;i++)
deap.emplace(nums[i],i);
answer.push_back(deap.top().first);
for(int i=k;i<nums.size();i++){
deap.emplace(nums[i],i);
while(deap.top().second<=i-k){
deap.pop();
}
answer.push_back(deap.top().first);
}
return answer;
}
};