4. Median of Two Sorted Arrays

最新推荐文章于 2023-06-12 16:30:30 发布

原创最新推荐文章于 2023-06-12 16:30:30 发布 · 144 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode

python 同时被 3 个专栏收录

20 篇文章

订阅专栏

leetcode

16 篇文章

订阅专栏

算法

15 篇文章

订阅专栏

本文介绍了一个算法问题：如何找到两个已排序数组的中位数，并确保整体运行时间复杂度为O(log(m+n))。通过将两个数组合并并排序的方法给出了问题的解答。

题目

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)).

You may assume nums1 and nums2 cannot be both empty.

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

思路

在这里插入图片描述

代码

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        a=[]
        for i in nums1:
            a.append(i)
        for i in nums2:
            a.append(i)
        a.sort()
        l=len(a)
        if l%2!=0:
            return float(a[l//2])
        else:
            return (a[l//2-1]+a[l//2])/2