LeetCode:239. Sliding Window Maximum

在这里插入图片描述

写这一题是因为这一题有特定的解法,记录一下防止忘记。

1.特定解法
针对这一题有一种专门的数据结构,叫做Monotonic Queue。一个deque,里面存放的是数组的索引,从前往后代表了下一个最大值的候选者。每次迭代到一个新的数组元素:1)看看需不需要淘汰最大的那个索引,因为它可能随着窗口的滑动而到外面去了。2)循环的将deque的尾元素代表的索引值对应的数组中的值和当前的那个值比较,如果当前的那个值大,就把尾元素弹出,再接着比较下一个尾元素,直到遇到比它大的或者将全部元素都弹出,再将当前的索引压入。3)因为deque的首元素代表的索引值对应的数组中的值是当前窗口中最大的,所以将对应元素压入返回的vector中。
其实直接看代码理解更简单

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> dq;
        vector<int> res;
        for (int i = 0; i < nums.size(); ++i) {
            if (!dq.empty() && dq.front() <= i-k) dq.pop_front();
            while (!dq.empty() && nums[dq.back()] <= nums[i])//这里小于或者小于等于都行
                dq.pop_back();
            dq.push_back(i);
            if (i >= k-1) res.push_back(nums[dq.front()]);//在形成窗口之后才压入最大值
        }
        return res;
    }
};

2.我自己想的一个方法
还算做了一定的优化,不是那种暴力的解法,仅供参考

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> ret;
        if (nums.empty() || k > nums.size())
            return ret;
        int maxIndex = findMax(nums, 0, k);
        int maxVal = nums[maxIndex];
        ret.push_back(maxVal);
        int remain = maxIndex - 0;
        for (int i = k; i < nums.size(); ++i) {
            if (nums[i] >= maxVal) {
                maxVal = nums[i];
                remain = k-1;
                ret.push_back(maxVal);
            } 
            else {
                --remain;
                if (remain < 0) {
                    maxIndex = findMax(nums, i-k+1, k);
                    maxVal = nums[maxIndex];
                    remain = maxIndex - (i-k+1);
                    ret.push_back(maxVal);
                }
                else ret.push_back(maxVal);
            }
        }
        return ret;
    }
private:
    int findMax(const vector<int> &nums, int index, int k) {
        int maxVal = INT_MIN;
        int ret = index;
        for (int i = index; i < index+k; ++i) {
            if (nums[i] > maxVal) {
                maxVal = nums[i];
                ret = i;
            }
        }
        return ret;
    }
};
posted @ 2018-12-13 19:30  于老师的父亲王老爷子  阅读(35)  评论(0)    收藏  举报