题目描述
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组 [2,3,4,2,6,2,5,1] 及滑动窗口的大小 3,那么一共存在 6 个滑动窗口,它们的最大值分别为 [4,4,6,6,6,5]。
注意:
数据保证 k 大于 0,且 k 小于等于数组长度。
样例
输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3
输出: [4, 4, 6, 6, 6, 5]
算法1
(暴力枚举) $O(n)$
C++ 代码
class Solution {
public:
vector<int> maxInWindows(vector<int>& nums, int k) {
vector<int> res;
int n = nums.size();
int max_num = INT_MIN, index = 0;
for(int i = 0; i < k; ++i){
if(max_num < nums[i]){
max_num = nums[i];
index = i;
}
}
res.push_back(index);
for(int i = 1, j = k; j < n; ++i, ++j){
if(i <= res.back()){
if(nums[j] > nums[res.back()]) res.push_back(j);
else res.push_back(res.back());
}
else{
int pt = i;
index = i;
max_num = nums[i];
while(pt <= j){
if(max_num < nums[pt]){
max_num = nums[pt];
index = pt;
}
++pt;
}
res.push_back(index);
}
}
for(int i = 0; i < res.size(); ++i){
res[i] = nums[res[i]];
}
return res;
}
};