给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。返回滑动窗口中的最大值。
进阶:
你能在线性时间复杂度内解决此题吗?
示例:
输入: 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
提示:
1 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
1 <= k <= nums.length
方法1:暴力法
在每个窗口内,找出最大值并写入ans
时间复杂度:O(nk)空间复杂度:O(1)
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k)
{
vector<int> ans;
for(int i=0;i+k<=nums.size();i++)
{
int maxn=INT_MIN;
for(int j=0;j<k;j++)
{
maxn=max(maxn,nums[i+j]);
}
ans.push_back(maxn);
}
return ans;
}
};
方法2:利用双端队列
首先,初始化一个双端队列存储最大值索引,在从0到k的范围内,使得储存的元素是递减的(但是并没有全部存储),这样最大值的索引就在队首,之后开始遍历数组,若队首元素已经出去,就从队首出队,队尾元素要小于之前的队尾,否则就先将队中小于该元素的从队尾出队,每次返回队首就是最大值。
时间复杂度:O(n)空间复杂度:O(k)
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k)
{
vector<int> ans;
deque<int> _deq;
for(int i=0;i<k;i++)
{
while(!_deq.empty()&&nums[i]>nums[_deq.back()])
{
_deq.pop_back();
}
_deq.push_back(i);
}
ans.push_back(nums[_deq.front()]);
for(int i=k;i<nums.size();i++)
{
if(!_deq.empty()&&_deq.front()<=i-k)
{
_deq.pop_front();
}
while(!_deq.empty()&&nums[_deq.back()]<nums[i])
{
_deq.pop_back();
}
_deq.push_back(i);
ans.push_back(nums[_deq.front()]);
}
return ans;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sliding-window-maximum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。