滑动窗口的最大值

Posted 2021-02-14 hesper

滑动窗口的最大值

题目描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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> ret;
        
        if ((num.size() >= size ) && (size > 0)) {
            deque<int> index;    // 用于存放下标的索引
            
            for (auto i = 0; i < size; i++) { // 存放前size个中的最大值
                // 重复元素只存储一份, 所以用大于等于号
                while ((! index.empty()) && num[index.back()] <= num[i])
                    index.pop_back();
                index.push_back(i);
            }
            
            for (auto i = size; i < num.size(); i++) {
                ret.push_back(num[index.front()]);
                // 后插入的元素值比前面的元素大时, 这个元素就是是窗里最大元素, 要把前面元素都弹出
                while ((!index.empty()) && num[index.back()] <= num[i])
                    index.pop_back();
                // 当前头部索引不在窗范围内时要弹出头部索引
                if ((!index.empty() && index.front() <= (i - size)))
                    index.pop_front();
                // 若后插入的元素比头部索引小时, 由于头部索引会移除; 移出后, 后插入的元素可能会是最大值
                index.push_back(i);
            }
            // 上一个fou循环中i指向的是窗的后一个元素, 当i=size()时,
            //最后一个窗内最大值并没有插入ret中
            ret.push_back(num[index.front()]);
        }
        return ret;
    }
};

比较简陋, 笨方法, 一次下移滑动窗口, 然后找最大值

class Solution {
public:
    int findMax(const vector<int>& vt, int pos, unsigned int num) {
        int max = vt[pos];
        
        for(auto i = 0; i < num; i++) {
            if (vt[pos+i] > max)
                max = vt[pos+i];
        }
        return max;
    }
    
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> ret;
        if (num.empty() || size < 1 || size > num.size())
            return ret;
        
        for (auto i = 0; i < (num.size() - size + 1); i++) {
            int temp = 0;
            temp = findMax(num, i, size);
            ret.push_back(temp);
        }
        return ret;
    }
};