Median of Two Sorted Arrays

2 篇文章 0 订阅
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;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值