滑动窗口的最大值

2017-08-30  本文已影响0人  单倍体

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。

这道题啊让我想就是先读入size个数字,然后比较一下然后找出最大值,之后就一步一步的把首部pop掉,从尾部push。每个过程呢,都查询一下pop的是不之前的最大值,如果不是就把之前的最大值和push进来的比较。产生新的最大值。如果pop的是最大值,那就在新的窗口里面在循环的找一遍,找出最大值。

而这道题的标签呢是栈和队列,可知这道题真正的目的是数据结构的使用。那我们从前pop在后面push,很容易想到的就是deque这个双端队列。那么这个方法呢倒不用我上面说的那么复杂。

代码:

class Solution
{
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> res;
        deque<int> s;//这个s里面存放的是num数组的下标
        for(unsigned int i=0;i<num.size();++i)
        {
            while(s.size()&& num[s.back()]<= num[i])
                s.pop_back();//当s里面有下标的时候比较一下这些下标位置的元素和当前元素的大小
            //因为我们是从s.size()==1 就开始这项操作,所以可以说s里面都是按照元素大小排序的下标,就是位于
            //s首部的当前最大元素的下表地址
            while(s.size() &&i-s.front()+1>size)
                s.pop_front();
            //刚才说过了,s只有一个元素,就是最大元素的下标。如果这个下表和当前位置的距离大于窗口
            //那我们是要删掉了
            s.push_back(i);
            if(size && i+1>size)
                res.push_back(num[s.front()]);//这个front下一轮就会被pop掉
        }
        return res;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读