描述
给定一个未排序的整数数组,找到其中位数。
中位数是排序后数组的中间值,如果数组的个数是偶数个,则返回排序后数组的第N/2个数。
样例
给出数组[4, 5, 1, 2, 3], 返回 3
给出数组[7, 9, 4, 5],返回 5
挑战
给定一个未排序的整数数组,找到其中位数。
中位数是排序后数组的中间值,如果数组的个数是偶数个,则返回排序后数组的第N/2个数。
样例
给出数组[4, 5, 1, 2, 3], 返回 3
给出数组[7, 9, 4, 5],返回 5
挑战
时间复杂度为O(n)
分析
先排序,再判断数组长度的奇偶性,最后返回中位数。这里,排序用到了sort函数,当数组的数比较少时,会用插入排序;当数组的数比较多时,会用归并排序,当数组数特别多时,会用快速排序。
程序
class Solution {
public:
/**
* @param nums: A list of integers
* @return: An integer denotes the middle number of the array
*/
int median(vector<int> &nums) {
// write your code here
sort(nums.begin(), nums.end());
int mid = nums.size() / 2;
if(nums.size() % 2 == 0)//偶数
return nums[mid - 1];
else//奇数
return nums[mid];
}
};