C++ 寻找两个正序数组的中位数

给定两个大小为 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

 

class Solution {
public:
    double findMedianSortedArrays(vector& nums1, vector& nums2) {
            if(nums1.size() > nums2.size()){
                return findMedianSortedArrays(nums2,nums1);
            }

            int m = nums1.size();
            int n = nums2.size();
            int left = 0,right=m,ansi = -1;
            int median1 =0,median2 = 0;

            while(left<=right){
                int i=(left + right)/2;
                int j=(m+n+1)/2 - i;
                
                int nums_im1 = (i==0?INT_MIN:nums1[i-1]);
                int nums_i = (i==m?INT_MAX:nums1[i]);
                int nums_jm1 = (j==0?INT_MIN:nums2[j-1]);
                int nums_j = (j==n?INT_MAX:nums2[j]);
                if(nums_im1<=nums_j){
                    ansi = i;
                    median1 = max(nums_im1,nums_jm1);
                    median2 = min(nums_i,nums_j);
                    left = i+1;
                }
                else{
                    right = i-1;
                }


            }

            return (m+n)%2 == 0?(median1+median2)/2.0:median1;

    }
};