480 Sliding Window Median 滑动窗口中位数

天大地大妈咪最大 提交于 2020-03-27 18:33:51

详见:https://leetcode.com/problems/sliding-window-median/description/

C++:

class Solution {
public:
    vector<double> medianSlidingWindow(vector<int>& nums, int k) 
    {
        vector<double> res;
        multiset<double> ms(nums.begin(), nums.begin() + k);
        auto mid = next(ms.begin(), k /  2);
        for (int i = k; ; ++i)
        {
            res.push_back((*mid + *prev(mid,  1 - k % 2)) / 2);        
            if (i == nums.size())
            {
                return res;
            }
            ms.insert(nums[i]);
            if (nums[i] < *mid)
            {
                --mid;
            }
            if (nums[i - k] <= *mid)
            {
                ++mid;
            }
            ms.erase(ms.lower_bound(nums[i - k]));
        }
    }
};

 参考:http://www.cnblogs.com/grandyang/p/6620334.html

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!