标签:
最近感觉自己需要让心情平静下来,慢慢做事情,做好实验室的事情保证自己的毕业,同时也要做好题目找到份好工作。
先从写博客开始吧,把做过的题目记下来,写写思路。
这道题我没有用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