码迷,mamicode.com
首页 > Windows程序 > 详细

lintcode Sliding Window Median

时间:2015-06-17 23:17:16      阅读:589      评论:0      收藏:0      [点我收藏+]

标签:

http://www.lintcode.com/en/problem/sliding-window-median/

 

最近感觉自己需要让心情平静下来,慢慢做事情,做好实验室的事情保证自己的毕业,同时也要做好题目找到份好工作。

先从写博客开始吧,把做过的题目记下来,写写思路。

 

这道题我没有用O(n log n)的方法,实际上应该是O(n log k),需要用两个set互相之间倒数据就可以了,判断起来比较烦。

直接用map也能够通过,线性查找sliding window的中间值,得到结果。每次进入新元素之前,都删掉前面的元素,做计数。

通过每一个元素的计数得到哪个值是中间值,同时也靠计数决定某个数是否被erase。

 

class Solution {
public:
    /**
     * @param nums: A list of integers.
     * @return: The median of the element inside the window at each moving
     */
    vector<int> medianSlidingWindow(vector<int> &nums, int k) {
        // write your code here
        map<int, int> rec;
        vector<int> res;
        
        if (k == 0 || nums.empty() || nums.size() < k)
            return res;
            
        int n = nums.size();
        for (int i = 0; i < k; ++i)
            ++rec[nums[i]];
        
        // 虽然不明白怎么回事但是AC了
        // 这部分再想想
        auto getMid = [&](){
            int target = k >> 1;
            target -= 1 - (k & 0x1);
            
            int cur = 0;
            auto iter = rec.begin();
            for ( ; iter != rec.end(); ++iter){
                cur += iter->second;
                if (cur > target) break;
            }
            
            return iter->first;
        };
        
        res.push_back(getMid());
        
        for (int i = k; i < n; ++i){
            auto iter = rec.find(nums[i - k]);
            --iter->second;
            if (iter->second == 0) rec.erase(iter);
            ++rec[nums[i]];
            
            res.push_back(getMid());
        }
        
        return res;
    }
};

 

lintcode Sliding Window Median

标签:

原文地址:http://www.cnblogs.com/EpisodeXI/p/4584549.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!