leetcode第4题python版寻找两个正序数组的中位数

最新推荐文章于 2023-01-06 22:46:04 发布

ICPunk

最新推荐文章于 2023-01-06 22:46:04 发布

阅读量1.4k

点赞数

分类专栏：算法文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/xinzaitt/article/details/123632565

版权

这篇博客介绍了如何找到两个正序数组的中位数，通过使用二分法将时间复杂度降低到O(log(min(len1,len2)))。算法首先确保较短的数组作为基准，并通过不断调整数组边界找到正确的位置。当数组长度为奇数时，返回左边最大值；为偶数时，返回左右两边最大和最小值的平均数。

摘要由CSDN通过智能技术生成

'''
4. 寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
参考思路：
https://www.bilibili.com/video/BV1H5411c7oC
'''
class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        # 1. 计算两个数组的长度
        len1 = len(nums1)
        len2 = len(nums2)

        # 2. 始终保证第一个数组是短数组，以短数组位计算基准，可以将计算复杂度降低到
        # O(log(min(len1,  len2))). 达到这种目有两种途径，一种是递归，一种直接数组位置互换
        if len1 > len2: 
            return self.findMedianSortedArrays(nums2, nums1) 
        
        # 3. 初始化短数组的二分法的区间范围及中位数，[0, len1], 左闭右闭，对应数组下标
        # 想象一下，i始终指向边界像右边那个值，极端的情况，nums数组边界右侧没有值时i=len1
        i_min, i_max, median = 0, len1, 0
        # 4. while循环、二分法处理短数组 易错点：i_min <= i_max, 包含等于
        while i_min <= i_max:
            # 4.1 根据二分法的思路初始化i的值
            i = (i_min + i_max) // 2