快乐的LeetCode --- 寻找两个正序数组的中位数

最新推荐文章于 2024-07-12 23:40:30 发布

原创最新推荐文章于 2024-07-12 23:40:30 发布 · 189 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

多味的LeetCode 专栏收录该内容

211 篇文章

订阅专栏

本文介绍了一种算法，用于找到两个已排序数组的中位数，通过合并并排序两个数组，然后根据数组长度的奇偶性计算中位数。提供三种Python代码实现方式，包括简化技巧。

摘要生成于 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):
        nums = []
        for i in range(len(nums1)):
            nums.append(nums1[i])
        for j in range(len(nums2)):
            nums.append(nums2[j])
        nums = sorted(nums)
        if len(nums) % 2 == 0:
            return (nums[len(nums)//2] + nums[(len(nums)//2)-1]) / 2.0
        else:
            return nums[len(nums)//2]

代码简化1：

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        nums1.extend(nums2)
        nums1 = sorted(nums1)
        if len(nums1) % 2 == 0:
            return (nums1[len(nums1)//2] + nums1[(len(nums1)//2)-1]) / 2.0
        else:
            return nums1[len(nums1)//2]

代码简化2：

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        nums = nums1 + nums2
        nums = sorted(nums)
        if len(nums) % 2 == 0:
            return (nums[len(nums)//2] + nums[(len(nums)//2)-1]) / 2.0
        else:
            return nums[len(nums)//2]