i 值用于切割nums1,j 用于切割num2
对 nums1数组 进行二分查找,搜寻到恰好符合要求的 i 值
要求: B[ j − 1 ] ≤ A[ i ] 且 A[ i - 1 ] ≤ B[ j ]
前提需要满足 :
1、Length(nums1) <= Length(nums2)
2、切割后左边累计数量 等于 右边累计数量: j = ( m + n + 1 ) / 2 - i (推导出的公式,当二分求出 i 值后,直接使用此公式求出 j 值即可满足要求)
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size();
int n = nums2.size();
if (m > n)
return findMedianSortedArrays(nums2, nums1);
int iBegin = 0, iEnd = m, halfLen = (m + n + 1) / 2;
while (iBegin <= iEnd) {
int i_1_mid = (iBegin + iEnd) / 2;
int j_2_mid = halfLen - i_1_mid;
if (i_1_mid < iEnd && nums2[j_2_mid - 1] > nums1[i_1_mid])
iBegin = i_1_mid + 1;
else if (i_1_mid > iBegin && nums1[i_1_mid - 1] > nums2[j_2_mid])
iEnd = i_1_mid - 1;
else {
int maxLeft = 0;
if (i_1_mid == 0)
maxLeft = nums2[j_2_mid - 1];
else if (j_2_mid == 0)
maxLeft = nums1[i_1_mid - 1];
else
maxLeft = max(nums1[i_1_mid - 1], nums2[j_2_mid - 1]);
if ((m + n) % 2 == 1)
return maxLeft;
int minRight = 0;
if (i_1_mid == m)
minRight = nums2[j_2_mid];
else if (j_2_mid == n)
minRight = nums1[i_1_mid];
else
minRight = min(nums2[j_2_mid], nums1[i_1_mid]);
return (static_cast<double>(maxLeft) + minRight) / 2.0f;
}
}
return 0.0f;
}
};