滑动窗口的最大值(C++)

题目:

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

思路:

使用双向队列,保存当前队列中的最大值位置,如果新滑入的值大于当前队列尾部的元素,则弹出队列尾部所有小于新值的元素,并把新滑入的值推入队列尾部;并判断队列头部保存的最大值的位置是否还在滑窗内,如果不在,则弹出该值;此时队列头部的值即为当前窗口最大值

代码实现:

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> result;
        int len = num.size();
        if(len >= size && size > 0)
        {
            deque<int> max_idx;
            for(int i = 0; i < size; i++)
            {
                while(!max_idx.empty() && num[i] >= num[max_idx.back()])
                    max_idx.pop_back(); // 弹出队列中小于当前值的所有数
                max_idx.push_back(i); 
            }
            for(int i = size; i < len; i++)
            {
                result.push_back(num[max_idx.front()]);
                while(!max_idx.empty() && num[i] >= num[max_idx.back()])
                    max_idx.pop_back();
                if(!max_idx.empty() && max_idx.front() <= i - size)
                    max_idx.pop_front(); // 保证队列中的最大位置在窗口内
                max_idx.push_back(i);
            }
            result.push_back(num[max_idx.front()]);
        }
        return result;
    }
};
发布了63 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Jerry_Leo_liu/article/details/105313150
今日推荐