leetcode热题系列之寻找两个正序数组的中位数（004）

leetcode热题系列第4题

4. 寻找两个正序数组的中位数

题目和示例:

给定两个大小为 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

思路：
1. 看到有序数组 log(m+n)，我们就要联想到二分法。
2. 我们可以将寻找中位数转换成另外一个问题：寻找topK问题。比如 [1,2,3,4] 中位数是2.5。也可以转换成：寻找TOP2和TOP3小问题
3. 由于题目是有两个数组。我们可以转换成以下思路解答：
每次寻找在两个数组找到 K/2小 的数字 比较大小。然后排除小的那组。然后继续迭代。如图：
（图源：详细通俗的思路分析，多解法 作者：windliang）

现在两个指针找到位置了。
比大小 显然 3 < 4 故 1 2 3 全部丢弃：
下一步迭代：

迭代到1 结束

还有一个终结条件：
比如1 3 4 9 已经舍去完了（长度为0） K仍不为1，这时候就直接返回剩下数组对应数字即可。

代码：

package Hot004;

public class Solution {

    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int n = nums1.length;
        int m = nums2.length;
        //编程技巧 这样就不用奇偶了 直接统一/2；
        int left = (n+m+1)/2;
        int right = (n+m+2)/2;

        return(getK(nums1,0,n-1, nums2,0,m-1, left) + getK(nums1,0,n-1, nums2,0,m-1, right))/2;

    }

    private double getK(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 getK(nums2,start2,end2,nums1,start1,end1,k);

        if(len1 == 0) return nums2[start1+k-1];


        if(k == 1) return Math.min(nums1[start1],nums2[start2]);

        int i = start1 + Math.min(len1,k/2)-1;
        int j = start2 + Math.min(len2,k/2)-1;

        if(nums1[i] <nums2[j]){
            return  getK(nums1,i+1,end1,nums2,start2,end2,k-(i-start1+1));
        } else return getK(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));

    }
}

github地址：https://github.com/CLFlemon/LeetcodeHot100