DataWhale:leetcode Day50

Chas_Liu

于 2019-06-28 09:02:54 发布

阅读量147

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/Chas_Liu/article/details/92831424

本文提供了一种解决LeetCode上两个有序数组的中位数问题的有效Python3代码实现。通过调整两个数组的指针位置，该算法能在O(log(min(m, n)))的时间复杂度内找到中位数，其中m和n分别是两个输入数组的长度。

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

一、leetcode地址

https://leetcode.com/problems/median-of-two-sorted-arrays/

二、问题描述

在这里插入图片描述

三、代码实现

语言：Python3
代码：

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m, n = len(nums1), len(nums2)
        if m > n:
            return self.findMedianSortedArrays(nums2,nums1)

        imin, imax, half = 0, m, (m + n + 1) // 2
        while imin <= imax:
            i = (imin + imax) // 2
            j = half - i
            if i < m and nums2[j-1] > nums1[i]:
                imin = i + 1
            elif i > 0 and nums1[i-1] > nums2[j]:
                imax = i - 1
            else:
                max_ = max(nums1[i-1] if i > 0 else float('-inf'),nums2[j-1] if j > 0 else float('-inf'))
                min_ = min(nums1[i] if i < m else float('inf'),nums2[j] if j < n else float('inf'))
                if (m + n) % 2 == 1:
                    return max_
                else:
                    return (max_ + min_) / 2