力扣第4题：寻找两个正序数组的中位数 库函数nth_element介绍（C++）

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

示例 

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

思路

nth_element是algorithm库下的一个函数，它的作用是：将区间中第n小的元素排在第n个位置上（升序排列）（注意不是整个区间排序，这个函数只保证第n个元素的正确）。

函数原型为:
void nth_element(iterator(begin),iterator(begin)+int,iterator(end));
例如 有数组a，则nth_element(a,a+2,a+4)会将前四个数中第3小的数排在第3位（
前四个数为a0,a1,a2,a3,a为第一个位置，a+2为第三个位置）

关于这个函数的内部详见：「数组」随机快速选择 / LeetCode LCR 076（C++）

解题方法

我们可以借助这个函数和简单的奇偶判断操作完成这个题。

利用vector的insert成员方法进行
nums1.insert(num1.end(),num2.begin(),num2.end()),将整个向量组2追加到向量1的后面，利用nth_element排中位数,判断向量长度的奇偶，选择直接输出中位数还是求平均值
实测击败90.5%。

Code

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        nums1.insert(nums1.end(),nums2.begin(),nums2.end());
        double ans;
        switch (nums1.size()%2){
            case 1:
            nth_element(nums1.begin(),nums1.begin()+nums1.size()/2,nums1.end());
            ans=nums1[nums1.size()/2];break;
            case 0:
            nth_element(nums1.begin(),nums1.begin()+nums1.size()/2,nums1.end());
            nth_element(nums1.begin(),nums1.begin()+nums1.size()/2-1,nums1.end());
            ans=(nums1[nums1.size()/2]+nums1[nums1.size()/2-1])/2.0;break;
        }
        return ans;
    }
};