【LeetCode第四题】寻找两个正序数组的中位数

---

题目：寻找两个正序数组的中位数

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

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

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

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

https://leetcode.cn/problems/median-of-two-sorted-arrays/

---

思路：合并 + sort

简单直接：合并 + sort
奇数：返回下标为 (m+n)/2的数
偶数：返回下标(m+n)/2 - 1 和 下标(m+n)/2的平均数

代码：击败了45.52%

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        nums1.insert(nums1.end(), nums2.begin(), nums2.end());
        sort(nums1.begin(), nums1.end());
        int length = nums1.size();
        int index = length / 2;
        int oddEven = length % 2;
        if (oddEven != 0) {
            return nums1[index];
        } else {
            return (nums1[index] + nums1[index - 1]) * 1.0 / 2;
        }
    }
};

总结

使用的是简单直接的：合并 + sort
当然效率更高的是双指针和二分查找思路：转化成寻找两个有序数组中的第 k 小的数，通过双指针，比较A[k/2 - 1] 和 B[k/2 - 1]，如果A[k/2 - 1] 小于 B[k/2 - 1]，A[0] 到 A[k/2−1]可以全部排除，一下排除了k/2个数。

谢谢观看，祝顺利！