lintcode-中位数-80

给定一个未排序的整数数组，找到其中位数。

中位数是排序后数组的中间值，如果数组的个数是偶数个，则返回排序后数组的第N/2个数。

样例

给出数组[4, 5, 1, 2, 3]， 返回 3

给出数组[7, 9, 4, 5]，返回 5

挑战 时间复杂度为O(n)

空间换时间：

class Solution {
public:
    
    int median(vector<int> &nums) {
        map<int,int> vis;
        int count;
        int minp=nums[0];
        int maxp=nums[0];
        
        for(auto e:nums){
            ++vis[e];      
            minp=min(e,minp);
            maxp=max(e,maxp);
        }
        int n=nums.size();
        count=(n&1)?n/2+1:n/2;
        for(int i=minp;i<=maxp;++i){
            while(vis[i]>0){
                --vis[i],--count;
                if(count==0)
                    return i;
            }
        }
    }
};