zoukankan      html  css  js  c++  java
  • 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);
            }
        }
    }
    
    
    
  • 相关阅读:
    Linux Exploit系列之一 典型的基于堆栈的缓冲区溢出
    [Codeforces Round #433][Codeforces 853C/854E. Boredom]
    Educational Codeforces Round 4
    [Educational Round 3][Codeforces 609F. Frogs and mosquitoes]
    [ACM-ICPC 2018 徐州赛区网络预赛][D. Easy Math]
    Educational Codeforces Round 50
    [Codeforces Round #507][Codeforces 1039C/1040E. Network Safety]
    [Educational Round 3][Codeforces 609E. Minimum spanning tree for each edge]
    Educational Codeforces Round 3
    [Manthan, Codefest 18][Codeforces 1037E. Trips]
  • 原文地址:https://www.cnblogs.com/disandafeier/p/12348290.html
Copyright © 2011-2022 走看看