Leetcode 4 寻找两个有序数组的中位数(还有一种解法没有学习)

题目

代码分析

要达到有log的时间复杂度,可以想到一定和二分有关系,我们的普通做法就是每次都从两个数组的头出去一个,当出去总量的一半之后,就完成了。
这里我们每次都出一半,具体的,如果两个数组的总量是14,我们要找了就是第7小和第8小的数字,对于第7小的,我们首先要找第7/2=3小的数字,这就是这道题的思路。例子:
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-w-2/
中的解法3

代码实现

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m=nums1.length;
        int n=nums2.length;

        int index1=(m+n+1)/2;
        int index2=(m+n+2)/2;

        return (f(nums1,0,m-1,nums2,0,n-1,index1)+f(nums1,0,m-1,nums2,0,n-1,index2))*0.5;
    }

    public int f(int[] nums1,int start1,int end1,int[] nums2,int start2,int end2,int k)
    {
         int len1=end1-start1+1;
         int len2=end2-start2+1;
         if(len1>len2)
         {
             return f(nums2,start2,end2,nums1,start1,end1,k);
         }
         if(len1==0)
         {
             return nums2[start2+k-1];   
         }
         if(k==1)
         {
             return Math.min(nums1[start1],nums2[start2]);
         }   
         int index1=start1+Math.min(k/2,len1)-1;
         int index2=start2+Math.min(k/2,len2)-1;
        if (nums1[index1] > nums2[index2]) {
            return f(nums1, start1, end1, nums2, index2 + 1, end2, k - index2 + start2 - 1);
        } else {
            return f(nums1, index1 + 1, end1, nums2, start2, end2, k - index1 + start1 - 1);
        }
    }
}

学习情况

1次

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务