滑动窗口问题

时间:2021-06-02 15:11:32

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

class SlideWindow {
public:
    vector<int> slide(vector<int> arr, int n, int w) {
        // write code here
        deque<int> max_index;
        int begin=0,end=0;
        vector<int> result;
        while(end<n)
        {
            if(!max_index.empty()&&max_index.front()<begin)
                max_index.pop_front();
            if(!max_index.empty()&&arr[max_index.front()]<arr[end])
            {
                max_index.clear();
                max_index.push_back(end);
            }
            else
            {
                while(!max_index.empty()&&arr[max_index.back()]<=arr[end])
                {
                    max_index.pop_back();
                }
                max_index.push_back(end);
            }
            if(end>=w-1)
            {
                result.push_back(arr[max_index.front()]);
            }
            end++;
            if(end>w-1)
            {
                begin++;
            }
        }
        return result;
    }
};