题目要求
给定两个大小为 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
思路解析
思路一
首先要知道中位数是什么,中位数,顾名思义,排序后最中间的数(若为偶数个,则为最中间的的俩数字的平均数)。
按照这个理解,很容易得到处理流程:
1.将两个数组合并然后排序
2.然后求中间的数,在根据奇偶性,返回不同的值
幸运的是,题目给定的两个有序数组,合并两个有序数组的经过优化后的时间复杂度大概是O(n),而题目要求的时间复杂度是O(log(m + n)),数据量较大的话,应该会不通过,那么这个思路在这道题目中显然不可行。另外,其实不用全部排序,排序数量达到元素总数量的1/2就行,这样也许能通过吧,不过我没有去试。
思路二
题目要求的时间负责度是O(log(m + n)),大概能猜到(显然?)需要使用二分去找中位数。
其次还是要知道中位数是什么,中位数需要满足一下三个要求:
a、中位数左边的都比中位数小
b、中位数右边的都比中位数大
c、中位数左右两边的元素数量相等
按照这样对位数的理解应该就相对容易去用二分实现了吧。
1、将第一个数组分成左右两部分,根据中位数要求c,第二个数组的分法也就确定了。
2、确定是否满足根据中位数要求a,b。也就是同时满足第一个数组的左边最大的数小于等于第二个数组右半部分,第二个数组的左边最大的数小于等于第一个数组右半部分。
3、若a、b不满足,需要判断第一个数组的正在切分位置是在当前切分位置的左边还是右边。
按照这个思路大概就能写出代码了把(写的不是很清楚,有点累了,改天完善一下),通过代码如下:
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
#define INTMAX 2147483647
#define INTMIN -2147483648
int eve = (nums1Size + nums2Size) / 2;
int isEven = 1 - (nums1Size + nums2Size) % 2;
int l = 0, r = nums1Size, mid1, mid2;
int max1, max2, min1, min2;
bool flag;
while (1)
{
mid1 = (l + r) / 2;//前面有mid1个
mid2 = eve - mid1;//前面有mid2个
mid2 = mid2 > nums2Size ? nums2Size : mid2;
mid2 = mid2 < 0 ? 0 : mid2;
max1 = (mid1 - 1) < 0 ? INTMIN : nums1[mid1 - 1];
max2 = (mid2 - 1) < 0 ? INTMIN : nums2[mid2 - 1];
min1 = mid1 >= nums1Size ? INTMAX : nums1[mid1];
min2 = mid2 >= nums2Size ? INTMAX : nums2[mid2];
flag = max1 <= min2 && max2 <= min1;
//std::cout<<mid1<<" "<<mid2<<std::endl;
// std::cout<<max1<<" "<<max2<<" "<<min1<<" "<<min2<<std::endl;
if ((mid1 + mid2) < eve)
{
l = mid1 + 1;
continue;
}
if ((mid1 + mid2) > eve)
{
r = mid1 - 1;
continue;
}
if (flag)
{
break;
}
if (max1 > min2)
{
r = mid1 - 1;
}
else
{
l = mid1 + 1;
}
}
max1 = max1 > max2 ? max1 : max2;
min1 = min1 < min2 ? min1 : min2;
//std::cout<< max1 << " " <<min1 <<" "<<isEven<<std::endl;
return (min1 * (2 - isEven) + max1 * isEven) / 2.0;
}