题目描述
给定两个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
若不考虑时间复杂度,我们可以将两个数组合并后进行排序,找第(m+n+1)/2位和第(m+n+2)/2位的元素,两者的平均数即为两个有序数组的中位数,然而,合并数组进行排序的时间复杂度为O(m+n),不符合题目要求。
此时,时间复杂度的要求为对数阶,因此我们想到用二分法,所谓的中位数是可以将一个有序集合分成两个相等长度的子集,题中规定的两个数组均为有序的,因此,我们需要在两个数组的左右两侧分别选出(m+n+1)/2个元素,并求出左子集的最大值与右子集的最小值的平均数,即为中位数。
代码
public class Solution {
public double FindMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.Length;
int n = nums2.Length;
if (m > n)
{
int[] change = nums1;
nums1=nums2;
nums2=change;
m=nums1.Length;
n=nums2.Length;
}
int k = (m + n + 1)/2;
int left = 0;
int right = m;
while (left < right)
{
int i = (left + right)/2;
int j = k - i;
if (nums1[i] < nums2[j - 1])
left = i + 1;
else
right = i;
}
int m1 = left;
int m2 = k - left;
int a=0,b=0;
if(m1!=0)
a=nums1[m1-1];
if(m2!=0)
b=nums2[m2-1];
int max = Math.Max(a,b);
if ((m + n)%2 == 1)
return max;
int a1=0,b1=0;
if(m1!=m)
a1=nums1[m1];
if(m2!=n)
b1=nums2[m2];
int min = Math.Min(a1,b1);
return (max + min)*0.5;
}
}