LeetCode 4 JAVA Median of Two Sorted Arrays

题目如下:

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

You may assume nums1 and nums2 cannot be both empty.

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

解法一:

比较容易想到的解法,先把两个数组合并成一个数组,然后用Java内置函数对数组进行排序,在进行数组中位数的输出,算法复杂度为O(n+m);代码如下:

class Solution {
      public double findMedianSortedArrays(int[] a, int[] b) {
        int[] c = new int[a.length+b.length];

        for (int i = 0; i < c.length; i++) {
            c[i] = i<a.length?a[i]:b[i-a.length];
        }//将两个数组合并成一个数组(不在乎数组速度),比较可取的方法

        Arrays.sort(c);

        if (c.length%2 == 1){
            return c[c.length/2]*1.0;
        }else {
            return (c[c.length/2]*1.0+c[c.length/2-1])/2;
        }
    }

}

解题思路:

 


  数组索引
  0
  1
  2
  3
  4
  5

 

  数组1(num1)
  3
  5
  8
  9
  
  


  数组2(num2)
  1
  2
  7
  10
  11
  12

 


其中,N1=4,N2=6,size=4+6=10.

1,现在有的是两个已经排好序的数组,结果是要找出这两个数组中间的数值,如果两个数组的元素个数为偶数,则输出的是中间两个元素的平均值。 
2,可以想象,如果将数组1随便切一刀(如在3和5之间切一刀),数组1将分成两份,数组1左别的元素的个数为1,右边的元素的个数为3。

由于数组1和数组2最终分成的左右两份的个数是确定的,都是所有元素的个数的一半(size/2=5)所以我们也可以知道,此时对数组2应该切的一刀的位置应该在10和11之间,数组2左边的个数为4,右边的个数为2.才能使两个数组左右两边的元素个数加起来的和(1+4=2+3)相等。 
另外,我们记在数组1靠近这一刀的左别的元素为L1(3),右边元素为R1(5).同理,记在数组2靠近这一刀的左别的元素为L2(10),右边元素为R2(11). 
如果这一刀的位置是正确的,则应该有的结果是 
L1<=R2 
L2<=R1 
这样就能确保,左边的元素都小于右边的元素了。

3,所以,我们只需要直接找出在数组1切这一刀的正确位置就可以了。 
为了减少查找次数,我们对短的数组进行二分查找。将在数组1切割的位置记为cut1,在数组2切割的位置记为cut2,cut2=(size/2)-cut1。 
cut1,cut2分别表示的是数组1,数组2左边的元素的个数。 
4,切这一刀的结果有三种 
1)L1>R2  则cut1应该向左移,才能使数组1较多的数被分配到右边。 
2)L2>R1 则cut1应该向右移,才能使数组1较多的数被分配到左边。 
3)其他情况(L1<=R2  L2<=R1),cut1的位置是正确的,可以停止查找,输出结果。 
5,其他说明 
1)考虑到边界条件,就是cut的位置可能在边缘,就是cut1=0或者cut1=N1,cut2=0或者cut2=N2的这些情况,我们将min和max两个特殊值分别加在数组1和数组2的两端,就可以统一考虑了。还有N1个数为0的时候,直接输出结果即可。 
2)为了减少查找时间,使用的是二分查找,就是cut1的位置是一半一半的查找的,实现时间只要log(N),不然就会超时。所以,我们不能只是简单地将cut1–或者cut1++,而是要记下每次cut1的区域范围,我们将cut1的范围记录下来,用[cutL,cutR]表示。一开始cut1的范围是[cutL,cutR]=[0,N1], 
如果L1>R2  则cut1应该向左移,才能使数组1较多的数被分配到右边。cut1的范围就变成了[cutL,cut1-1],下次的cut1的位置就是cut1 = (cutR - cutL) / 2 + cutL;。 
如果L2>R1 则cut1应该向右移,才能使数组1较多的数被分配到左边。cut1的范围就变成了[cut1+1,cutR],下次的cut1的位置就是cut1 = (cutR - cutL) / 2 + cutL;。 
3)数组的元素个数和是奇数的情况下,中间的元素应该就是min(R1,R2),只需另外处理输出就可以了。
 

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if(nums1.length>nums2.length)
           return findMedianSortedArrays(nums2,nums1);
         int MIN_VALUE = 0x80000000;
        int MAX_VALUE = 0x7fffffff;
        int cut1=0;
        int cut2=0;
        int cutl=0;
        int cutr=nums1.length;
        int len=nums1.length+nums2.length;
        
        while(cut1<=nums1.length){
            cut1=(cutr-cutl)/2+cutl;
            cut2=len/2-cut1;
            
            int l1= (cut1==0)?MIN_VALUE:nums1[cut1-1];
            int l2= (cut2==0)?MIN_VALUE:nums2[cut2-1];
            int r1= (cut1==0)?MAX_VALUE:nums1[cut1];
            int r2= (cut2==0)?MAX_VALUE:nums2[cut2];
            
            if(l1>r2)
                cutr=cut1-1;
            else if(l2<r1)
                cutl= cut1+1;
            else if(len%2 == 0){
                l1=l1<l2? l2 :l1;
                r1=r1<r2? r1 :r2;
                return (l1+r1)/2;
            }
               else{
                   r1=r1<r2? r1 :r2;
                   return r1;
               } 
                   
        }
        return -1;
    }
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值