给定两个大小分别为 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
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
解题思路:
二分查找
- 两个数组找第k个元素,a数组找第k/2个元素,b数组k/2个元素,如果a的大,那么b的前k/2个元素一定不包括第k个元素。如果相等,那么这就是第k个元素。
- 注意边界情况 如果其中一个数组为0,另一个数组的第k个就是答案,如果k==1,那么去两个数组当前第一个元素最小值
代码如下:
class Solution {
public:
//中位数 偶数 (m + n) /2 , (m + n) / 2 + 1
//nums1 取 k/2 ,nums2取 k/2 比较大小,如果一边已经不足,那么另一边第k个就是答案
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int k = nums1.size() + nums2.size();
int m = nums1.size();
int n = nums2.size();
if (k % 2 == 0) return (helper(nums1, 0, m, nums2, 0, n, k/2) + helper(nums1, 0, m, nums2, 0, n, k/2 +1) ) / 2.0;
else return helper(nums1, 0, m, nums2, 0, n, k/2 + 1);
}
double helper(vector<int>& nums1, int l1, int count1, vector<int>& nums2, int l2, int count2, int k) {
if (count2 < count1) return helper(nums2, l2, count2, nums1, l1, count1, k);
if (count1 == 0) return nums2[l2 + k -1];
if (k == 1) return min(nums1[l1], nums2[l2]);
int aoffset = min(count1, k / 2);
int boffset = k - aoffset;
if (nums1[l1 + aoffset - 1] < nums2[l2 + boffset - 1]) return helper(nums1, l1 + aoffset, count1 - aoffset, nums2, l2, count2, k - aoffset );
else if (nums1[l1 + aoffset - 1] > nums2[l2 + boffset - 1]) return helper(nums1,l1, count1, nums2, l2 + boffset, count2 - boffset, k - boffset);
else return nums1[l1 + aoffset -1];
}
};