lintcode-中位数-80

来源:互联网 发布:编程达人在线教育 编辑:程序博客网 时间:2024/04/29 08:28

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

中位数是排序后数组的中间值,如果数组的个数是偶数个,则返回排序后数组的第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;            }        }    }};


0 0
原创粉丝点击