两个有序数组的中位数

最新推荐文章于 2021-03-31 20:57:41 发布

竺一辉

最新推荐文章于 2021-03-31 20:57:41 发布

阅读量95

点赞数

本文链接：https://blog.csdn.net/SteveHui1995/article/details/104424771

版权

/*给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
*/

给出思路，但还是有一些不理解的地方。

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

    if (m > n) {
        return findMedianSortedArrays(nums2, nums1);
    }

    int i = 0, j = 0, imin = 0, imax = m, half = (m + n + 1) / 2;
    double maxLeft = 0, minRight = 0;
    while (imin <= imax) {
        i = (imin + imax) / 2;
        j = half - i;
        if (j > 0 && i < m && nums2[j - 1] > nums1[i]) {
            imin = i + 1;
        } else if (i > 0 && j < n && nums1[i - 1] > nums2[j]) {
            imax = i - 1;
        } else {
            if (i == 0) {
                maxLeft = (double) nums2[j - 1];
            } else if (j == 0) {
                maxLeft = (double) nums1[i - 1];
            } else {
                maxLeft = (double) Math.max(nums1[i - 1], nums2[j - 1]);
            }
            break;
        }
    }
    if ((m + n) % 2 == 1) {
        return maxLeft;
    }
    if (i == m) {
        minRight = (double) nums2[j];
    } else if (j == n) {
        minRight = (double) nums1[i];
    } else {
        minRight = (double) Math.min(nums1[i], nums2[j]);
    }

    return (double) (maxLeft + minRight) / 2;
}

}