lintcode Sliding Window Median

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;
    }
};