• 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;
        }
    };
  • 相关阅读:
    爬取阳光问政平台
    CrawlSpider爬取腾讯招聘信息
    LinkExtractor
    镜像源列表
    screen命令总结
    PHP解决中文乱码问题
    Linux查找大文件或目录
    CentOS6/7系列防火墙管理
    删除win10冗余的服务
    解决win10安装wireshark新版本,欢迎界面卡顿情况
  • 原文地址:https://www.cnblogs.com/EpisodeXI/p/4584549.html
Copyright © 2020-2023  润新知