经典力扣题：寻找两个正序数组的中位数

最新推荐文章于 2024-11-01 15:19:21 发布

SilenceFS

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量253

点赞数 6

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/SilenceFS/article/details/141821821

版权

一、问题

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

二、python解答

class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        A, B = nums1, nums2
        total = len(nums1) + len(nums2)
        half = total // 2
        
        if len(A) > len(B):
            A, B = B, A
        
        l, r = 0, len(A)
        while l <= r:
            i = (l + r) // 2
            j = half - i
            
            Aleft = A[i - 1] if i > 0 else float('-infinity')
            Aright = A[i] if i < len(A) else float('infinity')
            Bleft = B[j - 1] if j > 0 else float('-infinity')
            Bright = B[j] if j < len(B) else float('infinity')
            
            if Aleft <= Bright and Bleft <= Aright:
                if total % 2 == 0:
                    return (max(Aleft, Bleft) + min(Aright, Bright)) / 2
                else:
                    return min(Aright, Bright)
            elif Aleft > Bright:
                r = i - 1
            else:
                l = i + 1

        return None