004.median-of-two-sorted-arrays

最新推荐文章于 2022-09-18 18:35:40 发布

流动熵

最新推荐文章于 2022-09-18 18:35:40 发布

阅读量199

点赞数

CC 4.0 BY-SA版权

分类专栏： ProgrammingTopics 文章标签： leetcode 排序算法算法

本文链接：https://blog.csdn.net/m0_55389447/article/details/123035520

ProgrammingTopics 专栏收录该内容

127 篇文章

订阅专栏

这是一个关于算法的问题，目标是在两个已排序的数组nums1和nums2中找到它们合并后的中位数，要求整体运行时间复杂度为O(log(m+n))。示例给出了不同情况下的中位数计算方法，通过将较短的数组插入到较长数组的适当位置来找到中位数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        a, b = sorted((nums1, nums2), key=len)
        m, n = len(a), len(b)
        after = (m + n - 1) / 2
        lo, hi = 0, m
        while lo < hi:
            i = (lo + hi) / 2
            if after-i-1 < 0 or a[i] >= b[after-i-1]:
                hi = i
            else:
                lo = i + 1
        i = lo
        nextfew = sorted(a[i:i+2] + b[after-i:after-i+2])
        return (nextfew[0] + nextfew[1 - (m+n)%2]) / 2.0