题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。
牛客链接:
https://www.nowcoder.com/practice/1624bc35a45c42c0bc17d17fa0cba788?tpId=13&&tqId=11217&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
解题思路:
用两种方法解决:
第一种暴力解决,两层循环找到最大值
第二种采用双端队列,队列中的头节点保存的数据比后面的要大。比如当前假如的数据比队尾的数字大,说明当前这个数字最起码在从现在起到后面的过程中可能是最大值,而之前队尾的数字不可能最大了,所以要删除队尾元素。 此外,还要判断队头的元素是否超过size长度,由于存储的是下标,所以可以计算得到;在双端队列中保存的数字是传入的向量的下标;
代码一:
时间复杂度o(nk)
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> result;
int len = num.size();
if(len < size || size==0) return result;
int max = num[0];
for(int i=0; i<len-size+1; i++) {
max = num[i];
for(int j=i+size-1; j>i; j--) {
if(num[j] > max) {
max = num[j];
}
}
result.push_back(max);
}
return result;
}
};
代码二:
时间复杂度o(n),空间复杂度为o(n)
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> result;
int len = num.size();
if(len < size || size< 1) return result;
deque<int> maxIndex;
for(int i=0; i<size; i++) {
//当队列的末尾下标对应值小于当前下标对应值时,删除,一直循环
while(!maxIndex.empty() && num[i]>num[maxIndex.back()]) {
maxIndex.pop_back();
}
maxIndex.push_back(i);
}
for(int i=size; i<len; i++) {
result.push_back(num[maxIndex.front()]);
//当队列的末尾下标对应值小于当前下标对应值时,删除,一直循环
while(!maxIndex.empty() && num[i]>=num[maxIndex.back()]) {
maxIndex.pop_back();
}
//当队列的队首值已经超过滑动窗口大小,删除
if(!maxIndex.empty() && (int)(i-size)>=maxIndex.front()) {
maxIndex.pop_front();
}
maxIndex.push_back(i);
}
result.push_back(num[maxIndex.front()]);
return result;
}
};
注意:
双端队列