您的位置:首页 > 其它

Median of Two Sorted Arrays

2015-08-27 20:22 330 查看
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)).

解题思路:

先求出两个数组元素个数之和numsSize,则它们的中位数可以转换成求第k小的数

double func(int* nums1, int nums1Size, int* nums2, int nums2Size, int k)

if (nums1Size > nums2Size)
return func(nums2, nums2Size, nums1, nums1Size, k);
if (nums1Size == 0)
return nums2[k-1];
if (k == 1)
return (nums1[0] > nums2[0]) ? nums2[0] : nums1[0];

int pos_nums1 = (k/2) > nums1Size ? nums1Size : k/2;
int pos_nums2 = k - pos_nums1;
if (nums1[pos_nums1 - 1] > nums2[pos_nums2 - 1])
return func(nums1, nums1Size, nums2+pos_nums2, nums2Size-pos_nums2, k-pos_nums2);
if (nums1[pos_nums1 - 1] < nums2[pos_nums2 - 1])
return func(nums1+pos_nums1, nums1Size-pos_nums1, nums2, nums2Size, k-pos_nums1);
else
return nums1[pos_nums1-1];


}

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {

if (nums1 == NULL || nums2 == NULL)
return;

int numsSize = nums1Size+nums2Size;

if (numsSize & 0x01 == 1)//奇数
return func(nums1, nums1Size, nums2, nums2Size, numsSize/2+1);
else
return (func(nums1, nums1Size, nums2, nums2Size, numsSize/2) + func(nums1, nums1Size, nums2, nums2Size, numsSize/2+1))/2.0;


}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: