leetcode-腾讯精选练习（50 题）python #4 寻找两个有序数组的中位数

最新推荐文章于 2021-12-14 22:59:56 发布

楚天星云

最新推荐文章于 2021-12-14 22:59:56 发布

阅读量170

点赞数

分类专栏： leetcode-腾讯精选练习50题

本文链接：https://blog.csdn.net/weixin_43866606/article/details/93710879

版权

leetcode-腾讯精选练习50题专栏收录该内容

49 篇文章 1 订阅

订阅专栏

leetcode-腾讯精选练习（50 题）python #4 寻找两个有序数组的中位数
题目来源：https://leetcode-cn.com/problemset/50/
给定两个大小为 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:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        a, b, m = *sorted((nums1, nums2), key=len), (len(nums1) + len(nums2) - 1) // 2
        self.__class__.__getitem__ = lambda self, i: m-i-1 < 0 or a[i] >= b[m-i-1]
        i = bisect.bisect_left(self, True, 0, len(a))
        r = sorted(a[i:i+2] + b[m-i:m-i+2])
        return (r[0] + r[1 - (len(a) + len(b)) % 2]) / 2