leetcode480. 滑动窗口中位数

题目链接：https://leetcode-cn.com/problems/sliding-window-median/

题意：

中位数是有序序列最中间的那个数。如果序列的长度是偶数，则没有最中间的数；此时中位数是最中间的两个数的平均数。

例如：

[2,3,4]，中位数是 3
[2,3]，中位数是 (2 + 3) / 2 = 2.5
给你一个数组 nums，有一个长度为 k 的窗口从最左端滑动到最右端。窗口中有 k 个数，每次窗口向右移动 1 位。你的任务是找出每次窗口移动后得到的新窗口中元素的中位数，并输出由它们组成的数组。

方法：哈希表+集合 

class Solution {
public:
    vector<double> medianSlidingWindow(vector<int>& nums, int k) {
        vector<double> ret;
        multiset<double> st;//set和multiset都是有序的集合，底层都是基于红黑树实现的，区别是multiset允许重复
        int len = nums.size();//存储nums数组中的元素个数
        int flag = k%2;//判断k是属于奇数的还是偶数，偶数的话，中位数就是两个数取平均
        int center = k/2;//中间元素的位置
        for(int i=0;i<k;i++)//往st中插入元素
        {
            st.insert(nums[i]);
        }
        auto iter = st.begin();
        std::advance(iter,center);
        if(flag) ret.push_back(*iter);
        else
        {  
            ret.push_back((*iter+*prev(iter,1))/2);//prev(it)	it 为指定的迭代器，该函数默认可以返回一个指向上一个位置处的迭代器。注意，it 至少为双向迭代器。
        } 
        for(int i=1;i<len-k+1;i++)//滑动窗口滑动
        {
            st.insert(nums[i+k-1]);//插入最新的项
            st.erase(st.find(nums[i-1]));//删除滑动窗口左侧的那一项
            iter = st.begin();
            std::advance(iter,center);
            if(flag)
            {
                ret.push_back(*iter);
            } 
            else
            {  
                ret.push_back((*iter+*prev(iter,1))/2);
            } 
        }
        return ret;
    }
};

精简后：

class Solution {
public:
    vector<double> medianSlidingWindow(vector<int>& nums, int k) {
        vector<double> ret;
        multiset<double> st;//set和multiset都是有序的集合，底层都是基于红黑树实现的，区别是multiset允许重复
        int len = nums.size();//存储nums数组中的元素个数
        int center = k/2;//中间元素的位置
        for(int i=0;i<k;i++)//往st中插入元素
        {
            st.insert(nums[i]);
        }
        auto iter = st.begin();
        std::advance(iter,center);
        ret.push_back((*iter+*prev(iter,1-k%2))/2);//奇数就取中，偶数就取两个的平均
        for(int i=1;i<len-k+1;i++)//滑动窗口滑动
        {
            st.insert(nums[i+k-1]);//插入最新的项
            st.erase(st.find(nums[i-1]));//删除滑动窗口左侧的那一项
            iter = st.begin();
            std::advance(iter,center);
            ret.push_back((*iter+*prev(iter,1-k%2))/2);
        }
        return ret;
    }
};