题目
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
Example 1:
nums1 = [1, 3]
nums2 = [2]The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]The median is (2 + 3)/2 = 2.5
找两个有序数组合并后的中间数。
可以等价与找两个有序数组合并后的第k个数。
找第K小的数,可以将K平分到两个数组中,然后利用一个重要的结论:如果A[k/2-1] < B[k/2-1],那么A[0]~A[k/2-1]一定在第k小的数的序列当中,可以用反证法证明。
找第k个数代码:
假设数组1长度<数组2长度
double findKth(vector<int>& nums1, vector<int>& nums2, int k){
if (nums1.size() > nums2.size()){
return findKth(nums2, nums1, k);
}
if (nums1.size() == 0){
return nums2[k-1];
}
if (k == 1){
return std::min(nums1[0], nums2[0]);
}
int p1 = std::min(k/2, (int)nums1.size());
int p2 = k - p1; //共取k个数
if (nums1[p1-1] < nums2[p2-1]){
nums1.erase(nums1.begin(), nums1.begin() + p1);
return findKth(nums1, nums2, k - p1);
}
else if (nums1[p1-1] > nums2[p2-1]){
nums2.erase(nums2.begin(), nums2.begin() + p2);
return findKth(nums1, nums2, k - p2);
}
else {
return nums1[p1-1];
}
}