Sliding Window Problem Summary (Lintcode 360, Leetcode 239)

做Sliding Window的方法是:将其拆解为“加一个元素,和减一个元素”。注意,有的时候最好先减再加,有的时候最好先加再减。而另外一个点是:很多情况下,并不用分类讨论 window size >= array size的情况,如下面两题。

Lintcode 360:

Sliding Window找median。这道题最好先减再加,用的是基本的two priority queue的方法。

class Solution {
public:
    /**
     * @param nums: A list of integers.
     * @return: The median of the element inside the window at each moving
     */
    void insert(int num){
        if(m1.empty() || num <= *m1.begin()){
            m1.insert(num);
        }else{
            m2.insert(num);
        }
        balanceSet();
    } 
    
    void balanceSet(){
        if(m1.size() > m2.size() + 1){
            int cur = *m1.begin();
            m1.erase(m1.begin());
            m2.insert(cur);
        }
        if(m2.size() > m1.size()){
            int cur = *m2.begin();
            m2.erase(m2.begin());
            m1.insert(cur);
        }
    }
     
    void erase(int num){
        if(num <= *m1.begin()){
            m1.erase(m1.find(num));
        }else{
            m2.erase(m2.find(num));
        }
        balanceSet();
    } 
    vector<int> medianSlidingWindow(vector<int> &nums, int k) {
        // write your code here
        vector<int> ret;
        if(nums.empty() || k <= 0) return ret;
        int idx = min(k, (int)nums.size());
        for(int i=0; i<idx; i++){
            insert(nums[i]);
        }
        ret.push_back(*m1.begin());
        for(int i=idx; i<nums.size(); i++){
            erase(nums[i - k]);
            insert(nums[i]);
            ret.push_back(*m1.begin());
        }
        return ret;
    }
private:
    multiset<int, greater<int>> m1;
    multiset<int> m2;
};

Sliding Window Maxium:

这道题用Deque解,而最好先加后减的顺序。基本思路是:记录一个max,如果这个max即将被deque pop出去,则扫一遍deque,找新的max。

vector<int> maxSlidingWindow(vector<int>& nums, int k) {
    vector<int> ret;
    deque<int> dq;
    if(nums.empty() || k <= 0) return ret;
    int max_num = nums[0];
    for(int i=0; i<min(k, (int)nums.size()); i++){
        dq.push_back(nums[i]);
        if(nums[i] > max_num) max_num = nums[i];
    }
    ret.push_back(max_num);
    for(int i=k; i<nums.size(); i++){
        if(nums[i] >= max_num){
            max_num = nums[i];
            dq.push_back(nums[i]);
            dq.pop_front();
        }else{
            dq.push_back(nums[i]);
            int cur = dq.front(); dq.pop_front();
            if(cur == max_num){
                max_num = INT_MIN;
                for(auto it : dq){
                    if(it > max_num) max_num = it;
                }
            }
        }
        ret.push_back(max_num);
    }
    return ret;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,793评论 0 33
  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 18,036评论 2 36
  • 这类题是一类典型的two priority queue的题目。维护两个heap,一个MinHeap,一个MaxHe...
    stepsma阅读 397评论 0 0
  • 你的话让她哭泣我很难过因为,我心中有个孩子 你的态度让她失望我很难过因为,我心中有个孩子 你的强硬让她躲进房间我很...
    占从Zoey阅读 143评论 0 0
  • 古镇上的生活如同如山上流下的一股股清澈山泉,或激越或潺潺,只按它们自己的节奏和韵律,自成一首首或激荡或舒缓或轻快的...
    陌上花开hf阅读 419评论 4 9