【Leetcode-python】4.Median of Two Sorted Arrays(两个排序数组的中位数)

题目:有两个排序的数组nums1和nums2,长度分别为m和n。找到两个排序数组的中位数。总运行时间复杂度应为O(log(m + n))。可以假设nums1和nums2  不能都为空。
例:nums1 = [1,3] nums2 = [2]  中位数是2.0;nums1 = [1,2]  nums2 = [3,4]  中位数为(2 + 3)/ 2 = 2.5

思路:可以将两个数组列表组合形成一个新的列表,再对新的列表排序,如果(m+n)%2==0,则中位数为中间两个数的均值,如果(m+n)%2==1,则中位数为最中间一个数。

代码如下:
使用位移操作符 (m+n) >> 1比 int((m+n) /2) 慢

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m,n=len(nums1),len(nums2)
        list_merge=sorted(nums1+nums2)
        median_i=int((m+n)/2)
        if (m+n)%2==0:
             return (list_merge[median_i-1]+list_merge[median_i])/2
        else:
             return float(list_merge[median_i])
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值