给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
示例 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
题解一:
将两个数组合并,然后根据元素个数的奇偶性求中位数
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
{
double res = 0.0;
int len = nums1.size() + nums2.size();
vector<int> vec;
for (auto& x : nums1)
{
vec.push_back(x);
}
for (auto& x : nums2)
{
vec.push_back(x);
}
sort(vec.begin(), vec.end());
int m = vec.size() / 2;
if (vec.size() % 2 != 0)
{
res = vec[m];
}
else
{
res = (vec[m - 1] + vec[m]) / 2.0;
}
return res;
}