力扣：4 ，寻找两个正序数组的中位数

110 阅读 0 评论 73 点赞

我是靠谱客的博主自信画板，这篇文章主要介绍力扣：4 ，寻找两个正序数组的中位数，现在分享给大家，希望可以做个参考。

题目：

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

示例 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

示例 3：

输入：nums1 = [0,0], nums2 = [0,0]

输出：0.00000

示例 4：

输入：nums1 = [], nums2 = [1]

输出：1.00000

示例 5：

输入：nums1 = [2], nums2 = []

输出：2.00000

提示：

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

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

思路：将两个数组里的所有元素汇集到同一数组a中，再将a数组中的元素进行排序，最后判断a数组中的元素个数来决定中位数。

复制代码

1
2
3
4
5
6
7
8
9
10
11
class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> a;
        for (int i = 0; i < nums1.size(); ++i) a.push_back(nums1[i]);
        for (int i = 0; i < nums2.size(); ++i) a.push_back(nums2[i]);
        sort(a.begin(), a.end());
        if (a.size() & 1) return a[a.size() / 2];
        return double((a[a.size() / 2] + a[a.size() / 2 - 1])) / 2;
    }
};

ZHJ