1、题目描述
输入一个数组和一个k,k是移动窗口大小,向右移动窗口,找出每个窗口找那个的最大值。
2、思路
前找出前k个数的最大值,记录为m,并将其放入答案数组。
移动窗口,若最左移除的元素正好等于m,则需要重新求窗口的最大值,将其记录进m,并放入答案;
反之,更新m为当前元素和m的最大值,将m放入答案数组。
法2:用一个可在头尾插入删除的列表保存前k个数的最大值,保证列表的头元素数值最大,后面的递减。
每次插入一个新的元素,先判断要不要把列表头元素给删掉,再插入,返回的都是列表的头元素。
3、代码
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> ans;
int n =nums.size();
if(n==0) return ans;
int m = findmax(nums,0,k-1);
ans.push_back(m);
for(int i=k;i<n;i++){
if(nums[i-k]!=m){
m=max(m,nums[i]);
ans.push_back(m);
}
else{
m=findmax(nums,i-k+1,i);
ans.push_back(m);
}
}
return ans;
}
int findmax(vector<int>& nums, int i,int j){
int m=nums[i];
for(int k=i+1;k<=j;k++)
m=max(m,nums[k]);
return m;
}
法2:
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> ans;
int n = nums.size();
if(n==0) return ans;
vector<int> q;
for(int i = 0; i < k; i++) {
while(q.size() > 0 && nums[i] > nums[q[q.size()-1]])
q.pop_back();
q.push_back(i);
}
ans.push_back(nums[q[0]]);
for(int i = k; i < n; i++) {
if(q[0] < i - k + 1)
q.erase(q.begin());
while(q.size() > 0 && nums[i] > nums[q[q.size()-1]])
q.pop_back();
q.push_back(i);
ans.push_back(nums[q[0]]);
}
return ans;
}
};