Median of Two Sorted Arrays

There are two sorted arraysnums1andnums2of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

思路:
设m = nums1.length, n = nums2.length, 假设m < n(大于就交换过来)
设min = 0, max = m, i = (min + max)/2, j = (m + n)/2 - i(中位数在[0, (n+m)/2]之间),
比较nums1[i-1],nums[i]和nums2[j-1], nums2[j]
得到正确解的情况是:nums2[j-1] < nums1[i]
得到的结果要分为奇偶讨论.
另外对于其他两种情况:
nums1[i] < nums2[j-1]表明i太小需要将i增大,即修改min = i+1, 这种情况要确保i != m <==> j > 0;
nums1[i-1] > nums2[j]表明i太大需要将i减小,即修改max = i -1, 这种情况要确保i > 0 <==> j < n;
另外需要处理边界情况
当i=0时
max_left = nums2[j-1]
当j=0时
max_left = nums1[i-1]
否则max_left = max(nums1[i-1], nums2[j-1])
当i=m时
min_right = nums2[j]
当j=n时
min_right = nums1[i]
否则
min_right = min(nums1[i], nums2[j])
当数组长度为奇数时:
result = max(nums1[i-1], nums2[j-1])
当数组长度为偶数时:
result = (max_left + min_right )/2;
java代码:

public double findMedianSortedArrays(int[] nums1, int[] nums2) {

    int m=nums1.length,n=nums2.length;

    if(m>n){
        return findMedianSortedArrays(nums2, nums1);
    }

    int mid = (m+n+1)/2;
    int minPos = 0;
    int maxPos = m;
    int pos1 = (maxPos+minPos)/2;
    int pos2 = mid - pos1;
    int num1=0,num2=0;
    
    while(minPos<=maxPos){
        
        if(pos1>0 && nums1[pos1-1]>nums2[pos2]){
            maxPos = pos1-1;
        }else if(pos1 < m && nums1[pos1]<nums2[pos2-1]){
            minPos = pos1+1;
        }else {
            if(pos1==0){
                num1=nums2[pos2-1];
            }else if(pos2==0){
                num1=nums1[pos1-1];
            }else{
                num1 = Math.max(nums1[pos1-1], nums2[pos2-1]);
            }
            break;
        }
        pos1 = (minPos+maxPos)/2;
        pos2 = mid - pos1;
    }
    
    if((m+n)%2==1)return num1;
    
    if(pos1==m){
        num2=nums2[pos2];
    }else if(pos2 == n){
        num2=nums1[pos1];
    }else {
        num2=Math.min(nums1[pos1], nums2[pos2]);
    }
 
    return (num1+num2)/2.0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容