码迷,mamicode.com
首页 > 编程语言 > 详细

LeetCode寻找两个有序数组的中位数

时间:2019-08-31 12:33:48      阅读:55      评论:0      收藏:0      [点我收藏+]

标签:假设   amp   中位数   商业   class   return   sorted   max   ted   

题目:

给定两个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
 
方法:二分法
将数组A左边i个元素和数组B右边j个元素
使得i+j=(m-i)+(n-j)
 数组A剩下的len1-i和数组B剩下的len2-j个元素之和保持一致
此时只要满足条件a[i-1]<b[j]&&b[j-1]<a[i]时,中位数就可得
处理奇偶时,为保证i+j=(m+n)/2=>i+j=(m+n-1)/2 偶数=>不变 奇数平半分
因为不能直接改变i和j所有用start和end来控制,或者说是i的范围
int m=nums1.length,n=nums2.length;
        if(m>n)//保证len1<=len2
            return getZ(nums2,nums1);
        
        int start=0,end=m;
        int maxl = 0,minr = 0;
        
        while(start<=end) {
            int i=(start+end)/2;
            int j=(m+n+1)/2-i;
            
            if(i>start&&nums1[i-1]>nums2[j]) {//i太大应该变小,右边范围左移
                end--;
            }else if(i<end&&nums2[j-1]>nums1[i]){//i太小应该变大,左边范围扩大
                start++;
            }else {
                maxl=(i==0)?nums2[j-1]:((j==0)?nums1[i-1]:(nums1[i-1]>nums2[j-1]?nums1[i-1]:nums2[j-1]));
                if((m+n)%2!=0)
                    return maxl;
                else
                    minr=(i==m)?nums2[j]:((j==n)?nums1[i]:(nums1[i]<nums2[j]?nums1[i]:nums2[j]));
                    return (maxl+minr)/2.0;
            }
        }
        return 0.0;

过程中遇到的问题及理解:

1,红色第一处以及红色第二处如果规定为(i>=1)(i<=m)执行时间会变长,或者是说start和end为什么可以代表i的范围

2,此处的返回不能写在外面因为当m=0时,j可能会越界(如{}{1}这两个数组执行时),因此奇数时直接返回,而只有奇数时才会越界

3,end--和start++的确定不太明白,还是想不通start和end为什么可以代表i的范围

LeetCode寻找两个有序数组的中位数

标签:假设   amp   中位数   商业   class   return   sorted   max   ted   

原文地址:https://www.cnblogs.com/code-fun/p/11438417.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!