题目描述
给定两个大小分别为 m 和 n 的升序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个升序数组的中位数 。
注意 : 1 <= m+n <= 2000, 即两个数组不同时为空。
解题思路
此题暴力解法为合并两个数组,重新排序,返回中间位置的数,即为中位数。此处不讨论此方法。
另一个思路是不用合并两个数组,直接找到中位数的位置,由于两个数组长度已知,因此中位数的位置也是已知的((m+n)/ 2);可以维护两个指针,初始化执行数组起始位置,每次将较小值的指针后移一位(若一个指针已到达尾部,需要移动另一个指针),循环(m+n)/ 2次,即是中位数的位置。
解题代码
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
int len = nums1Size + nums2Size;
int left = -1, right = -1;
int nums1Index = 0, nums2Index = 0;
for (int i=0; i<=len/2; ++i)
{
left = right;
if (nums1Index < nums1Size &&
(nums2Index >= nums2Size || nums1[nums1Index] < nums2[nums2Index]))
{
right = nums1[nums1Index++];
}
else
{
right = nums2[nums2Index++];
}
}
return (len & 1) == 0 ? (left + right) / 2.0 : right;
}