1、问题描述
给定一个未排序的整数数组,找到其中位数。
中位数是排序后数组的中间值,如果数组的个数是偶数个,则返回排序后数组的第N/2个数。
给出数组[4, 5, 1, 2, 3], 返回 3
给出数组[7, 9, 4, 5],返回 5
2、实现思路
先用sort()函数排序,数组元素个数为偶数返回nums[(n-1)/2],数组元素个数为奇数返回nums[n/2]。
3、代码
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 n=nums.size();
if(n%2!=0) return nums[n/2];
else return nums[(n-1)/2];
}
};
4、感想
先用sort()函数排序,数组元素个数为偶数返回nums[(n-1)/2],数组元素个数为奇数返回nums[n/2]。