力扣leetcode题库练习（C++）—4. 寻找两个有序数组的中位数

题目描述

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例

示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0

示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5

分析

1.中位数定义：
数组按大小排列，若长度是奇数，则中位数为最中间的数值，若长度是偶数，则中位数为最中间的两个数值的平均数。
2.可以直接将两个数组合并成一个数组，进行排序，按定义计算出中位数。

代码及注释如下

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {//定义两个数组
        int a = nums1.size(),b=nums2.size();//两个数组的大小
        if(a < b){
            swap(nums1,nums2);//交换两个数组            
        }
        for(int i= 0;i < nums2.size();i++)
        {
            nums1.push_back(nums2[i]);//将nums2里的数放进nums1的后面
        }
        sort(nums1.begin(),nums1.end());//排序，从小到大
        double result;
        if(nums1.size() % 2 == 1)//如果是总数为奇数
            result = nums1[(nums1.size()+1)/2 - 1];
        else//如果为偶数
            result = (nums1[(nums1.size()/2) -1] +  nums1[(nums1.size()/2)]) / 2.0;
        return result;
    }
};

结果如下

注

1.swap(a,b)为交换函数，交换a和b的值。
2.nums1.push_back(a),意思是将a插入nums1数组的后面
3.sort(start,end,cmp)
（1）start表示要排序数组的起始地址；
（2）end表示数组结束地址的下一位；
（3）cmp用于规定排序的方法，可不填，默认升序。
如果想降序，可定义一个比较函数compare()

bool compare(int a,int b)
{   
return a>b;
}
sort(start,end,compare)