滑动窗口练习

来源:互联网 发布:杭州淘宝服装摄影 编辑:程序博客网 时间:2024/06/05 17:59

有一个整型数组 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]

滑动窗口问题,利用双端队列deque,保留窗口范围内最大值下标和最大值后面的最大值坐标(不好解释可以理解为在窗口内维持单调关系) 时间复杂度可以做到O(n)

class SlideWindow {public:    vector<int> slide(vector<int> arr, int n, int w) {        vector<int> res;        deque<int> deq;        for(int i=0;i!=n;++i)            {            while(1){            if(deq.empty()){                deq.push_back(i);                break;            }            else if(arr[i]<arr[deq.back()])                {                deq.push_back(i);                break;            }            else                deq.pop_back();                }            if(i>=w-1)                {                while(deq.front()<i-w+1)                    deq.pop_front();                res.push_back(arr[deq.front()]);            }                }                return res;           }};
0 0
原创粉丝点击