Lintcode——中位数



1.题目

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

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

样例

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

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

2.思路

       对数组进行排序，再取排在中间位置的数。

       注意1：数组长度为奇数和偶数，取中位数的区别。注意2：数组元素下标从0开始。

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)-1];
        else
           return nums[N/2];
    }
};

4.感想

        一开始看到这个题目觉得挺简单的，结果写完之后总是和预期结果差一位数。。一下子恍然大悟自己犯了低级的错误，数组元素下标从零开始，要返回排序后数组的第N/2个数时，要写return nums[(N/2)-1];