题目:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。你可以假设 nums1 和 nums2 不会同时为空。
思路:
忘数组中加入虚拟#,长度为0~2*n共 2*n+1,让长度恒为奇数,然后还有割点的概念,这个操作太妙,解决了长度为奇数偶数的问题,不用分情况讨论了。
参考:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/4-xun-zhao-liang-ge-you-xu-shu-zu-de-zhong-wei-shu/
题解:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
int m = nums2.size();
// 让小的为num1
if(n > m){
return findMedianSortedArrays(nums2,nums1);
}
int start = 0 , end = 2*n;// 加入虚拟#,长度为0~2*n共 2*n+1
int c1,c2;//数组1,2的割点
int LMax1,LMax2,RMin1,RMin2,LMax,RMin; //割点左右的值
while(start <= end){
c1 = (start + end)/2;
c2 = m + n - c1;
//给割点左右的值赋值,注意处理边界情况
LMax1 = (c1 == 0)?INT_MIN:nums1[(c1-1)/2];
RMin1 = (c1 == 2*n)?INT_MAX:nums1[c1/2];
LMax2 = (c2 == 0)?INT_MIN:nums2[(c2-1)/2];
RMin2 = (c2 == 2*m)?INT_MAX:nums2[c2/2];
if(LMax2 > RMin1)
//在数组一的割点的右边寻找新割点
start = c1 + 1;
else if(LMax1 >RMin2)
end = c1 - 1;
else
break;
}
//循环完就得到正确的 LMax 和 RMax值
LMax = (LMax1>LMax2)?LMax1:LMax2;
RMin = (RMin1>RMin2)?RMin2:RMin1;
return (LMax + RMin)/2.0;
}
};