滑动窗口最大值的C++实现
滑动窗口是一种常见的算法,它可以在一个数组或序列上移动一个固定大小的窗口,以便在这个窗口内进行一些操作。例如,可以求出在一个数组中大小为k的滑动窗口内的最大值。
在本文中,我们将探讨如何使用C++实现滑动窗口最大值算法。
算法思路
滑动窗口最大值算法的基本思路是维护一个大小为k的双端队列deque。在遍历数组时,对于每个元素,将其加入队列尾部,并检查队列头部元素是否已经超出窗口范围。如果是,则将队列头部元素弹出。
然后,从队列尾部开始,将所有小于当前元素的元素弹出。这是因为这些元素已经不可能成为窗口内的最大值了。最后,将当前元素加入队列尾部。
此时,队列头部元素就是当前窗口内的最大值。将其记录下来,并继续遍历数组直到遍历结束。
代码实现
下面是使用C++实现滑动窗口最大值算法的代码:
```
#include <iostream>
#include <deque>
#include <vector>
using namespace std;
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> dq;
vector<int> res;
for(int i = 0; i < nums.size(); i++) {
if(!dq.empty() && dq.front() == i - k) {
dq.pop_front();
}
while(!dq.empty() && nums[dq.back()] < nums[i]) {
dq.pop_back();
}
dq.push_back(i);
if(i >= k - 1) {
res.push_back(nums[dq.front()]);
}
}
return res;
}
int main() {
vector<int> nums = {1,3,-1,-3,5,3,6,7};
int k = 3;
vector<int> res = maxSlidingWindow(nums, k);
for(int i = 0; i < res.size(); i++) {
cout << res[i] << " ";
}
return 0;
}
```
在上面的代码中,我们定义了一个双端队列dq和一个结果向量res。在遍历数组时,我们首先检查队列头部元素是否已经超出窗口范围,如果是,则将其弹出。
然后,我们从队列尾部开始,将所有小于当前元素的元素弹出。最后,将当前元素加入队列尾部。
如果当前元素的下标大于等于k-1,则说明当前窗口内已经有k个元素了。此时,队列头部元素就是当前窗口内的最大值。将其记录下来,并将其加入结果向量res中。
最后,返回结果向量res即可。
测试结果
我们使用上面的代码对数组{1,3,-1,-3,5,3,6,7}和k=3进行测试,得到的结果如下:
```
3 3 5 5 6 7
```
可以看到,我们得到了正确的结果。滑动窗口最大值算法可以很方便地解决一些问题,例如找到一个数组中所有大小为k的子数组中的最大值。
总结
本文介绍了如何使用C++实现滑动窗口最大值算法。该算法可以很方便地找到一个数组中大小为k的滑动窗口内的最大值。通过维护一个双端队列,我们可以在O(n)的时间复杂度内解决这个问题。