leetcode --4.寻找两个正序数组的中位数 python

最新推荐文章于 2022-07-09 20:57:02 发布

weixin_45739347

最新推荐文章于 2022-07-09 20:57:02 发布

阅读量264

点赞数

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

本文链接：https://blog.csdn.net/weixin_45739347/article/details/113382808

版权

这篇博客介绍了如何解决LeetCode的第4题，即寻找两个正序数组的中位数。作者提供了两种方法，一是使用双指针策略，通过比较数组元素来移动指针；二是转换为寻找第K小数的问题，通过剔除较小的元素来定位中位数。

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

leecode - 4.寻找两正序数组的中位数

在这里插入图片描述
方法一：双指针
使用两个指针p1, p2分别指向nums1和nums2，进行比较，如果nums1[p1] < nums2[p2]，则p1+=1，相反则p2 += 1

方法二：第K小个数
将找中位数问题转化为寻找第K小个数的问题
k = (len(nums1)+len(nums2)) // 2

每次剔除一个数组里面较小的 k // 2 个数

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        length = len(nums1) + len(nums2)
        if len(nums1) < len(nums2): # 长序列的放在前面，防止越界
            nums1, nums2 = nums2, nums1
        
        if length == 2:
            if len(nums2) == 0: # 中位数在n