【LeetCode】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

---

解题
寻找中位数的问题可以化为寻找第k小值问题。例如：长度为9的数组的中位数等于数组中第5小的数，长度为8的数组的中位数等于数组中第4小的数与第5小的数的和的一半。

这里主要的难点在于我们通常都是解一个数组的中位数问题，这里变成了两个数组，而且时间复杂度限制在O(log (m+n)) 。我们要找到第k小的数，可以先把小于第k小的数的数给排除掉，然后要获取到这个第k小的数就简单了。这些小于第k小的数的数分布于两个数组中，所以我们可以把k分为 i = k/2 和 j = k - i 两部分。当nums1[i - 1] < nums2[j - 1]说明nums1中前i个数都小于两数组合并后的第k小的数，可以排除掉，接着我们继续找两数组合并后的第k-i小的数。反过来同理。当两个数相等时，这个数便是我们要找的第k小的数。

这里还有几个限制条件：
1、当nums1的长度为0时，直接返回nums2中第k小的数

2、当k==1时，说明我们要取第1小的数，直接返回nums1[0]和nums2[0]中较小的一个

---

代码

import java.util.Arrays;
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int length = nums1.length + nums2.length;
        if (length % 2 != 0){
            return findKth(nums1, nums2, (length + 1) / 2);
        }else{
            return (findKth(nums1, nums2, (length + 1 ) / 2) + findKth(nums1, nums2, (length + 2) / 2)) / 2.0;
        }
    }

    private int findKth(int[] nums1, int[] nums2, int k) {
        int m = nums1.length;
        int n = nums2.length;
        //令nums1长度小于nums2长度
        if (m > n) return findKth(nums2, nums1, k);
        //如果nums1长度为0,直接返回nums2中第k小的数
        if (m == 0) return nums2[k - 1];
        //如果k等于1,直接返回nums1[0]和nums2[0]中较小的一个
        if (k == 1) return Math.min(nums1[0], nums2[0]);

        int i = Math.min(m, k / 2);
        int j = Math.min(n, k - i);
        //如果nums1[i - 1] < nums2[j - 1]说明nums1中前i个数都小于两数组合并后的第k小的数,可以除去,继续找两数组合并后的第k-i小的数
        if (nums1[i - 1] < nums2[j - 1]) return findKth(Arrays.copyOfRange(nums1, i, m), nums2, k - i);
        //如果nums1[i - 1] > nums2[j - 1]说明nums2中前j个数都小于两数组合并后的第k小的数,可以除去,继续找两数组合并后的第k-j小的数
        if (nums1[i - 1] > nums2[j - 1]) return findKth(nums1, Arrays.copyOfRange(nums2, j, n), k -j);
        //相等的话这个数就是两数组合并后的第k小的数
        return nums1[i - 1];
    }
}