原文链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
图示(来源):
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
if(nums1.size()>nums2.size()){//保证nums1长度小于nums2,否则交换
vector<int> tmp=nums2;
nums2=nums1;
nums1=tmp;
}
int m=nums1.size();
int n=nums2.size();
int totalLeft=(m+n+1)/2;//中位线的左半部分
int left=0;
int right=m;
while(left<right){
int i=left+(right-left+1)/2;//nums1的中位线索引
int j=totalLeft-i;//nums2的中位线索引
if(nums1[i-1]>nums2[j]){//如果成立则表示nums1中位线左边的值大于nums2中位线右边的值,需要调整位置
right=i-1;
}else{
left=i;
}
}
int i=left;
int j=totalLeft-i;
int nums1MaxLeft=i==0?INT_MIN:nums1[i-1];//nums1中位线左边的值
int nums1MinRight=i==m?INT_MAX:nums1[i];//nums1中位线右边的值
int nums2MaxLeft=j==0?INT_MIN:nums2[j-1];//nums2中位线左边的值
int nums2MinRight=j==n?INT_MAX:nums2[j];//nums2中位线右边的值
if((m+n)%2){//如果是奇数,直接取中位线左边那个最大的
return max(nums1MaxLeft,nums2MaxLeft);
}else{//如果是偶数,取中位线两边的数的平均数
return ((double)(max(nums1MaxLeft,nums2MaxLeft)+min(nums1MinRight,nums2MinRight)))/2.0;
}
}