题目描述
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。你可以假设 nums1 和 nums2 不会同时为空。
解题思路
这题是一道很好的分治思想的题目,因为这两个数组是有序的,然后题目的要求是对数复杂度,我们很容易想到分治/二分的做法。求这个总数组的中位数,可以分别求出两个有序数组的中位数,我们设这两个中位数分别为 m i d 1 mid_1 mid1和 m i d 2 mid_2 mid2,求出后我们可以比较这两个数的大小,如果这个数组的中位数比较大,说明真正的中位数在这个中位数的前面或者在那个较小的中位数的后面,这个时候我们递归求解剩下的问题就行
参考代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1 = nums1.length;
int len2 = nums2.length;
int k = len1 + len2;
if(k % 2 == 1)
return getpositionval(nums1, 0, nums2, 0, k / 2 + 1);
else
return (getpositionval(nums1, 0, nums2, 0, k / 2) + getpositionval(nums1, 0, nums2, 0, k / 2 + 1)) / 2.0;
}
//获得指定顺序位置上的值,使用二分方法,k为顺序
public int getpositionval(int[] nums1, int head1, int[] nums2, int head2, int k){
if(head1 >= nums1.length)
return nums2[head2 + k - 1];
if(head2 >= nums2.length)
return nums1[head1 + k - 1];
if(k == 1)
return Math.min(nums1[head1], nums2[head2]);
int val1 = 0X3FFFFFFF;
int val2 = 0x3FFFFFFF;
if(head1 + k / 2 - 1 < nums1.length)
val1 = nums1[head1 + k / 2 - 1];
if(head2 + k / 2 - 1 < nums2.length)
val2 = nums2[head2 + k / 2 - 1];
if(val1 < val2)
return getpositionval(nums1, head1 + k / 2, nums2, head2, k - k / 2);
else
return getpositionval(nums1, head1, nums2, head2 + k / 2, k - k / 2);
}
}