Median of Two Sorted Arrays

来源:互联网 发布:批量查询域名是否备案 编辑:程序博客网 时间:2024/06/06 05:00

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;

}

0 0