Leetcode每日必刷题库第4题，如何寻找两个正序数组的中位数？

最新推荐文章于 2022-11-03 13:38:54 发布

文宇肃然

最新推荐文章于 2022-11-03 13:38:54 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：大厂面经及leetcode解题思路详细讲解

本文链接：https://blog.csdn.net/wenyusuran/article/details/107463151

大厂面经及leetcode解题思路详细讲解专栏收录该内容

36 篇文章 ¥19.90 ¥99.00

订阅专栏

本文探讨了如何在O(log(m + n))的时间复杂度内找到两个正序数组nums1和nums2的中位数。通过示例1（nums1 = [1, 3], nums2 = [2]，中位数为2.0）和示例2（nums1 = [1, 2], nums2 = [3, 4]，中位数为2.5），解释了算法的应用。" 28014945,1423167,理解与实现：树形选择排序,"['算法', '数据结构', '排序算法']

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

题目：

给定两个大小为 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

答案：

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

了解本专栏