double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
if (nums1Size == 0 && nums2Size == 0) return -1;
else if (nums1Size == 1 && nums2Size == 1)
{
return (nums1[0] + nums2[0]) / 2.0;
}
else if (nums1Size == 0)
{
if (nums2Size % 2 == 1)
{
return nums2[nums2Size / 2];
}
else
{
return (nums2[nums2Size / 2] + nums2[nums2Size / 2 - 1]) / 2.0;
}
}
else if (nums2Size == 0)
{
if (nums1Size % 2 == 1)
{
return nums1[nums1Size / 2];
}
else
{
return (nums1[nums1Size / 2] + nums1[nums1Size / 2 - 1]) / 2.0;
}
}
else
{
if (nums1Size == 2 && nums2Size % 2 == 0)
{
if (nums1[0] > nums2[nums2Size / 2 - 1] && nums1[1] < nums2[nums2Size / 2])
{
return (nums1[0] + nums1[1]) / 2.0;
}
}
if (nums2Size == 2 && nums1Size % 2 == 0)
{
if (nums2[0] > nums1[nums1Size / 2 - 1] && nums2[1] < nums1[nums1Size / 2])
{
return (nums2[0] + nums2[1]) / 2.0;
}
}
double mid1Data, mid2Data;
if (nums1Size % 2 == 1)
{
mid1Data = nums1[nums1Size / 2];
}
else
{
mid1Data = (nums1[nums1Size / 2] + nums1[nums1Size / 2 - 1]) / 2.0;
}
if (nums2Size % 2 == 1)
{
mid2Data = nums2[nums2Size / 2];
}
else
{
mid2Data = (nums2[nums2Size / 2] + nums2[nums2Size / 2 - 1]) / 2.0;
}
if (mid1Data == mid2Data) return mid1Data;
int cutSize1, cutSize2, cutSize;
if (nums1Size % 2 == 0)
{
cutSize1 = nums1Size / 2 - 1;
}
else
{
cutSize1 = nums1Size / 2;
}
if (nums2Size % 2 == 0)
{
cutSize2 = nums2Size / 2 - 1;
}
else
{
cutSize2 = nums2Size / 2;
}
cutSize = (cutSize1 <= cutSize2) ? cutSize1 : cutSize2;
if (nums1Size == 2 || nums2Size == 2)
{
cutSize = 1;
}
if (nums1Size == 1)
{
if (nums2Size % 2 == 0)
{
if (nums1[0] < nums2[nums2Size / 2 - 1])
{
return nums2[nums2Size / 2 - 1];
}
else if (nums1[0] > nums2[nums2Size / 2])
{
return nums2[nums2Size / 2];
}
else
{
return nums1[0];
}
}
else
{
if (nums1[0] < nums2[nums2Size / 2 -1 ])
{
return (nums2[nums2Size / 2] + nums2[nums2Size / 2 - 1]) / 2.0;
}
else if (nums1[0] > nums2[nums2Size / 2 + 1])
{
return (nums2[nums2Size / 2] + nums2[nums2Size / 2 + 1]) / 2.0;
}
else
{
return (nums1[0] + nums2[nums2Size / 2]) / 2.0;
}
}
}
else if (nums2Size == 1)
{
if (nums1Size % 2 == 0)
{
if (nums2[0] < nums1[nums1Size / 2 - 1])
{
return nums1[nums1Size / 2 - 1];
}
else if (nums2[0] > nums1[nums1Size / 2])
{
return nums1[nums1Size / 2];
}
else
{
return nums2[0];
}
}
else
{
if (nums2[0] < nums1[nums1Size / 2 - 1])
{
return (nums1[nums1Size / 2] + nums1[nums1Size / 2 - 1]) / 2.0;
}
else if (nums2[0] > nums1[nums1Size / 2 + 1])
{
return (nums1[nums1Size / 2] + nums1[nums1Size / 2 + 1]) / 2.0;
}
else
{
return (nums2[0] + nums1[nums1Size / 2]) / 2.0;
}
}
}
if (mid1Data < mid2Data)
{
return findMedianSortedArrays(&nums1[cutSize], nums1Size - cutSize, nums2, nums2Size - cutSize);
}
else
{
return findMedianSortedArrays(nums1, nums1Size - cutSize, &nums2[cutSize], nums2Size - cutSize);
}
}
}
4. Median of Two Sorted Arrays
最新推荐文章于 2024-09-22 04:59:47 发布