leetcode #4 寻找两个有序数组的中位数

题目

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例

思路

1. 观察时间复杂度，应该采用二用二分法
2. 两个数组分别二分，总体的中位数肯定是在二分后4块数据中的两块
3. 其实是对中位数的位置K二分

4.参考 https://www.cnblogs.com/grand...

代码

JAVA

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int left = (m + n + 1)/2;
        int right = (m + n + 2)/2;
        return (findKth(nums1, 0, nums2, 0, left)  + findKth(nums1, 0, nums2, 0, right))/2.0;
            
        
    }
    int findKth(int[] nums1, int i, int[] nums2, int j, int k){
        if(i >= nums1.length) return nums2[j + k - 1];
        if(j >= nums2.length) return nums1[i + k - 1];
        if(k == 1) return Math.min(nums1[i], nums2[j]);
        int midVal1 = (i + k/2 - 1 < nums1.length)?nums1[i + k/2 - 1]:Integer.MAX_VALUE;
        int midVal2 = (j + k/2 - 1 < nums2.length)?nums2[j + k/2 - 1]:Integer.MAX_VALUE;
        if(midVal1 < midVal2)
            return findKth(nums1,i + k/2, nums2, j, k - k/2);
        else 
            return findKth(nums1, i, nums2, j + k/2, k - k/2);
    }
}