LeetCode--4. 寻找两个有序数组的中位数(Median of Two Sorted Arrays)

题目描述

给定两个大小为 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

思路

题解和题目描述

class Solution {
public:
double findMedianSortedArrays(vector<int> num1, vector<int> num2) {
int m=num1.size();
int n=num2.size();
double ans;
if(m>n)
{//确保要从个数少的数组里划分，否则会造成越界
vector<int> temp(num1);
num1.swap(num2);
num2.swap(temp);
int tempp=m;
m=n;
n=tempp;
}
int imin=0,imax=m,halfl=(m+n+1)/2;
//i为划分的位置,取值为0-m,halfl为总体小的分组，且个数跟大的分组相等，或多1个
while(imin<=imax)
{
int i=(imin+imax)/2;
int j=halfl-i;
if(i<imax&&num1[i]<num2[j-1])
imin=i+1;    //i需要再大点
else if(i>imin&&num1[i-1]>num2[j])
imax=i-1;   //i需要再小点
else
{//i已经达到要求
int ansl=0;
if(i==0)//说明num1数组在小数组部分没有数，故小数组最大值在num2中找
ansl=num2[j-1];
else if(j==0)//同上
ansl=num1[i-1];
else //说明num1和num2在小数组中都有数,所以取两个中较大的数
ansl=num1[i-1]>num2[j-1]?num1[i-1]:num2[j-1];

if((n+m)&1)
{
ans=ansl;
break;
}

int ansr=0;
//说明num1数组的所有数都在小数组中
//故大数组最小值只能在num2中
if(i==m)
ansr=num2[j];
else if(j==n)//同
ansr=num1[i];
else
ansr=num1[i]<num2[j]?num1[i]:num2[j];
ans=(ansl+ansr)/2.0;
break;
}
}
return ans;
}
};