- 寻找两个正序数组的中位数
给定两个大小为 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
double findMedianSortedArrays(int* nums1, int size1, int* nums2, int size2){
int len = size1 + size2;
int p1 = 0, p2 = 0;
int pre = -1, next = -1;
for(int i = 0; i <= len / 2; i++){
pre = next;
if(p1 < size1 && (p2 >= size2 || nums1[p1] < nums2[p2]))
next = nums1[p1++];
else
next = nums2[p2++];
}
if(len % 2 == 0)
return (pre + next) / 2.0;
return next;
}