4. Median of Two Sorted Arrays

最新推荐文章于 2024-04-05 16:34:21 发布

jason_mai

最新推荐文章于 2024-04-05 16:34:21 发布

阅读量138

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： Leetcode

本文链接：https://blog.csdn.net/qq_36267931/article/details/80115802

Leetcode 专栏收录该内容

15 篇文章

订阅专栏

本文探讨了如何找出两个已排序数组的中位数，并提出了一种简单但有效的解决方案，即合并并排序两个数组，然后找到中位数。尽管这种方法的时间复杂度略高于题目要求，但它易于理解和实现。

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

题目链接https://leetcode.com/problems/median-of-two-sorted-arrays/description/

题目描述

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

题目大意

给出两个排序链表，给出两个链表合并后的中位数，时间复杂度应该为 O(log (m+n))。

分析

这道题的难点在于时间复杂度的要求。直接想出时间复杂度为O(log (m+n))的算法有点难度，先尝试一种简单的方法：

把两个链表连接起来，再排序，输出中位数即可，主要时间消耗在排序上，网上查阅可知链表的排序算法时间复杂度为O((m+n)*log(m+n)),虽然没有符合题目意思，也Accepted了。在Discuss模块中有大神提供了一种O(log (min(m,n)))的方法https://leetcode.com/problems/median-of-two-sorted-arrays/discuss/2481/Share-my-O(log(min(mn))-solution-with-explanation

我的解法

class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        num=nums1+nums2
        num.sort()
        # print((num))
        numlen=len(num)
        if (numlen==1):
            return num[0]
        elif (numlen==2):
            return (num[0]+num[1])*0.5
        elif (numlen%2==0):
            return (num[int(numlen/2)]+num[int(numlen/2-1)])*0.5
        else:
            return num[int(numlen/2)]