给定一个未排序的整数数组,找到其中位数。
中位数是排序后数组的中间值,如果数组的个数是偶数个,则返回排序后数组的第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;
}
}
}
};