题目链接:力扣
题意:给定一个数组
nums
和滑动窗口的大小k
,请找出所有滑动窗口里的最大值。
方法一: 可重复集合,时间复杂度O(nlogn),logn来自mst的排序
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
if(nums.size()==0) return res;
multiset<int> mst;
int size = nums.size();
k = k<size?k:size;
for(int i=0;i<k;i++)
{
mst.insert(nums[i]);
}
res.emplace_back(*mst.rbegin());//取mst的末端元素
for(int i=0;i<size-k;i++)
{
mst.erase(mst.find(nums[i]));//删除头元素
mst.insert(nums[i+k]);//插入元素
res.emplace_back(*mst.rbegin());
}
return res;
}
};
方法二:暴力,时间复杂度O(nk)
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
if(nums.size()==0) return res;
int start = 0,end = k-1;
//暴力
int size = nums.size();
while(end<size)
{
int t = INT_MIN;
start = end-(k-1);
for(int i = start;i<=end;i++)
{
t = max(t,nums[i]);
}
res.emplace_back(t);
end++;//右侧自加
}
return res;
}
};
方法三:优先队列,时间复杂度O(nlogn),logn来自于优先队列自身的排序
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
if(nums.size()==0) return {};
vector<int> res;
int size = nums.size();
priority_queue<pair<int,int>> q;
for(int i=0;i<k;i++)
{
q.emplace(nums[i],i);//首先是第一个滑动窗口
}
res.emplace_back(q.top().first);//将堆顶的元素放入res
for(int i=k;i<size;i++)
{
q.emplace(nums[i],i);
while(q.top().second<=i-k)//核心代码
{
q.pop();//堆顶元素属于已经滑动过的,就删掉它
}
res.emplace_back(q.top().first);
}
return res;
}
};
方法四:双端队列,时间复杂度O(n)
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int size = nums.size();
if(size==0) return {};
vector<int> res;
deque<int> dq;
// 双端队列,确保双端队列严格递增
for(int i=0;i<k;i++)
{
while(!dq.empty()&&nums[i]>=nums[dq.back()])
{
dq.pop_back();//弹出队尾元素
}
dq.push_back(i);
}
res.emplace_back(nums[dq.front()]);
for(int i=k;i<size;i++)
{
while(!dq.empty()&&nums[i]>=nums[dq.back()])//核心代码
{
dq.pop_back();//弹出队尾元素,确保双端队列严格递增
}
dq.push_back(i);
while(dq.front()<=i-k)//核心代码
{
dq.pop_front();//弹出队首一些超过范围的数
}
res.emplace_back(nums[dq.front()]);
}
return res;
}
};