给定两个大小分别为 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
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
与合并两个有序数组相似,两个数组不会同时为空,所以如果一个为空直接去另外一个数组找中位数,如果都不为空就比较两个数组的当前元素,那个元素小就合并到新的数组并把数组的指针后移继续比较。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int l1=nums1.size(),l2=nums2.size(),s=l1+l2;
if(l1==0){
if(l2%2==1)return nums2[l2/2];
else return double((nums2[l2/2]+nums2[l2/2-1]))/2;
}
if(l2==0){
if(l1%2==1)return nums1[l1/2];
else return double((nums1[l1/2]+nums1[l1/2-1]))/2;
}
int mid=s/2,flag=1;
vector<int> re;
if(s%2==0){
flag=0;
}
int count=0,i=0,j=0;
while(count<=mid){
if(i<l1&&j<l2){
if(nums1[i]<nums2[j]){
re.push_back(nums1[i]);
i++;
count++;
}
else {
re.push_back(nums2[j]);
j++;
count++;
}
}
else{
if(j<l2){
re.push_back(nums2[j]);
j++;
count++;
}
else {
re.push_back(nums1[i]);
i++;
count++;
}
}
}
if(flag)return re[mid];
else return double((re[mid-1]+re[mid]))/2;
}
};