• 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

    题解:
    首先求中位数,两种情况:
    (1)假设两个数组的总长度是奇数, 那么其中位数是一个值
    (2)假设偶数的中位数是两个值的平均值
    这样子就要分奇数跟偶数长度来算了。我们探索到一个规律,
    假设两个数组和起来长度是偶数10,那么就是 第5个跟第6个位置的数了,5 = (10+1)/2, 6 = (10 + 2 )/ 2;
    假设两个数组和起来长度是奇数11, 那么中位数就是第6个位置的数了, 6 = (11 + 1) / 2, 6 = (11 + 2) / 2;
    所以题目可以转化成第 (n + 1) / 2 跟 (n + 2) / 2位上的数的平均值。

    接着,我们的问题转移成:求两个有序数组中第n大的位置上的值了。

    class Solution {
        public double findMedianSortedArrays(int[] nums1, int[] nums2) {
            int n = nums1.length + nums2.length;
            int a = (n + 1) / 2;
            int b = (n + 2) / 2;
            search(nums1, nums2, 0, 0, a);
            search(nums1, nums2, 0, 0, b);
            return (double)(search(nums1, nums2, 0, 0, a) + search(nums1, nums2, 0, 0, b)) / 2;
        }
    
        public int search(int[] nums1, int[] nums2, int index1, int index2, int target) {
            if (index1 >= nums1.length) {
                return nums2[index2 + target - 1];
            }
            if (index2 >= nums2.length) {
                return nums1[index1 + target - 1];
            }
    
            if (target == 1) {
                return Math.min(nums1[index1], nums2[index2]);
            }
            int midVal1 = (index1 + target / 2 - 1 < nums1.length) ? nums1[index1 + target / 2 - 1] : Integer.MAX_VALUE;
            int midVal2 = (index2 + target / 2 - 1 < nums2.length) ? nums2[index2 + target / 2 - 1] : Integer.MAX_VALUE;
            if (midVal1 < midVal2) {
                return search(nums1, nums2, index1 + target / 2, index2, target - target / 2);
            } else {
                return search(nums1, nums2, index1, index2 + target / 2, target - target / 2);
            }
        }
    }
    
    
    
  • 相关阅读:
    poj 1511Invitation Cards
    hust 1608Dating With Girls
    sdibt 2128Problem A:Convolution Codes
    hdu 1325Is It A Tree?
    poj 2240Arbitrage
    hdu 2818Building Block
    poj 1789Truck History
    poj 1125Stockbroker Grapevine
    展望未来
    告别过去
  • 原文地址:https://www.cnblogs.com/disandafeier/p/12348290.html
Copyright © 2020-2023  润新知