题目:
有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7],w=3为例。因为第一个窗口[4,3,5]的最大值为5,第二个窗口[3,5,4]的最大值为5,第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。
给定整形数组arr及它的大小n,同时给定w,请返回res数组。保证w小于等于n,同时保证数组大小小于等于500。
测试样例:
[4,3,5,4,3,3,6,7],8,3
返回:[5,5,5,4,6,7]
基本思想:
- 定义一个双端队列qmax,然后依次遍历整形数组arr的每个数arr[i]。
- 如果当前qmax为空,则直接将当前遍历的arr[i]的下标i放到qmax的队尾。
- 如果当前qmax不为空,则比较arr[i]与qmax的队尾下标在arr数组对应的值arr[j]的大小。
- 如果
arr[i]<arr[j]
,则直接把当前遍历的arr[i]的下标i放到qmax的队尾。 - 如果
arr[i]>=arr[j]
,则依次将qmax对尾下标依次弹出,直到arr[i]<arr[j]
,再重复步骤4. - 这样在遍历的过程中,qmax队首的值就一直是滑动窗口的最大值。如果出现qmax队首的值滑出滑动窗口,则将qmax队首的值弹出。
代码如下:
class SlideWindow {
public:
vector<int> slide(vector<int> arr, int n, int w) {
// write code here
vector<int> res;
deque<int> qmax;
for(int i=0;i<n;i++)
{
while(!qmax.empty()&&arr[qmax.back()]<=arr[i])//对应步骤5
qmax.pop_back();
qmax.push_back(i);//对应步骤4
if(i+1>=w)//当滑动窗口出现
{
while(qmax.front()<i-w+1)//当滑动窗口过期
qmax.pop_front();
res.push_back(arr[qmax.front()]);//加入滑动窗口最大值
}
}
return res;
}
};