题目:
给定两个大小分别为 m
和 n
的正序(从小到大)数组 nums1
和 nums2
。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n))
。
示例:
- 输入: n u m s 1 = [ 1 , 3 ] , n u m s 2 = [ 2 ] nums1 = [1,3], nums2 = [2] nums1=[1,3],nums2=[2]
- 输出: 2.00000 2.00000 2.00000
- 解释: 合并数组 = [1,2,3] ,中位数 2
解题思路:
定义数组总长度int tot = nums1.size() + nums2.size()
,判断数组为奇数还是偶数,若为奇数,则找到第tot / 2 + 1
大的数返回;若为偶数,则找到第tot / 2
大和第tot / 2 + 1
大的数返回(left + right) / 2.0
。
然后定义查找算法(递归)。假设nums1
数组短,若它较长,则交换。然后判断k
的大小,若k==1
代表两数组最多各只剩一个数,先判断短的数组nums1
是否为空,若为空的话直接返回nums2
的数组,否则返回两个数组的最小值。若nums1
递归到空时,则直接返回nums2
的第k
个数。前面的操作都在定义终止条件,接着进行递归判断,定义si
和sj
为两个数组第k/2
个元素的下一个位置。然后比较两数组第k/2
元素值得大小,若nums1
大,则nums2
中前k/2
个元素一定小于第k
个数,可以舍去,否则舍去num1中前k/2
个元素。以此递归,方可解题。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int tot = nums1.size() + nums2.size();
if (tot % 2 == 0) {
int left = find(nums1, 0, nums2, 0, tot / 2);
int right = find(nums1, 0, nums2, 0, tot / 2 + 1);
return (left + right) / 2.0;
} else {
return find(nums1, 0, nums2, 0, tot / 2 + 1);
}
}
int find(vector<int>& nums1, int i, vector<int>& nums2, int j, int k){
// 假设nums1短,若nums1长的话,那么交换一下,先算nums2
if (nums1.size() - i > nums2.size() - j) return find(nums2, j, nums1, i, k);
// 当只剩一个数时,代表找到了最终的两个数
if (k == 1) {
// 若nums1为空的话,则返回nums2的那个数
if (nums1.size() == i) return nums2[j];
// 否则返回两者的最小值
else return min(nums1[i], nums2[j]);
}
// 若nums1为空了,返回nums2的第k个数
if (nums1.size() == i) return nums2[j + k - 1];
// 两个数组的第k/2个元素的下一个元素的位置
int si = min((int) nums1.size(), i + k / 2), sj = j + k - k / 2;
// 若nums1的第k/2个元素值大于nums2的第k/2的元素值,那么nums2的前半段无用
if (nums1[si - 1] > nums2[sj - 1]) return find(nums1, i, nums2, sj, k - (sj - j));
else return find(nums1, si, nums2, j, k - (si - i));
}
};