给你一个整数数组?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
#include <iostream>
#include <vector>
#include <deque>
#include <queue>
using namespace std;
int main()
{
vector<int> nums = { 1,3,-1,-3,5,3,6,7 };
int k = 3;
priority_queue<pair<int, int>> q;///用堆是因为 最大元素会排列在第一位,pop时也是第一个出来
for (int i = 0; i < k; ++i) {
q.emplace(nums[i], i);
}
vector<int> ans = { q.top().first };
for (int i = k; i <nums.size() ; ++i) {
q.emplace(nums[i], i);
while (q.top().second <= i - k) {
q.pop();
}
ans.push_back(q.top().first);
}
}