【重点】【二分查找】4.寻找两个正序数组的中位数

原创

已于 2025-08-23 17:01:47 修改 · 515 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#二分查找 #寻找两个正序数组的中位数

于 2023-12-24 15:34:19 首次发布

文章介绍了如何使用二分查找算法在两个已排序数组中找到合并后的中位数，时间复杂度为O(log(min(m,n)))，空间复杂度为O(1)。详细描述了一个名为Solution的类中的findMedianSortedArrays方法实现过程。

题目

最佳解法：二分查找

在这里插入图片描述

# 写法1
class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        if len(nums1) > len(nums2):
            nums1, nums2 = nums2, nums1
        m, n = len(nums1), len(nums2)
        iMin, iMax, half = 0, m, (m+n+1)//2
        while iMin <= iMax:
            i = (iMin + iMax) // 2  # nums1中分割线左侧元素数量
            j = half - i            # nums2中分割线左侧元素数量
            if i < m and nums2[j-1] > nums1[i]: # i需要增大
                iMin += 1
            elif i > 0 and nums1[i-1] > nums2[j]:
                iMin -= 1
            else:
                nums1LMax = -inf if i == 0 else nums1[i-1]
                nums1RMin = inf if i == m else nums1[i]
                nums2LMax = -inf if j == 0 else nums2[j-1]
                nums2RMin = inf if j == n else nums2[j]
                leftMax = max(nums1LMax, nums2LMax)