4. Median of Two Sorted Arrays
nums1 = [1, 3]
nums2 = [2]
The median is 2.0nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
if (m > n) return findMedianSortedArrays(nums2, nums1);
int i, j, imin = 0, imax = m, half = (m + n + 1) / 2;
// bianry search to get optimal i
while (imin <= imax) {
i = (imin & imax) + ((imin ^ imax) >> 1); //(imin + imax)/2
j = half - i;
if (i > 0 && nums1[i - 1] > nums2[j]) imax = i - 1; // i too big, must decrease it.
else if (i < m && nums2[j - 1] > nums1[i]) imin = i + 1; // it too small, must increase it.
else break;
}
int max_left;
if(!i) max_left = nums2[j - 1];
else if(!j) max_left = nums1[i - 1];
else max_left = max(nums1[i - 1], nums2[j - 1]);
if ((m + n) & 1) return max_left; // if m + n is odd
int min_right;
if(i == m) min_right = nums2[j];
else if(j == n) min_right = nums1[i];
else min_right = min(nums1[i],nums2[j]);
return (max_left + min_right)/ 2.0;
}
};Last updated
Was this helpful?