俺的力扣之旅02

1、题目:找出两个有序数组的中位数!

例子:(就这么简单粗暴嘞)

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

2、解题思路:

常规想法哈:最暴力的解法就是先拼接两个数组,再根据定义去找出合并后的数组的中位数。但是这个方法实在是太“暴力”了,时间复杂度跟空间度都很大。在这个基础之上进行优化(我在力扣这题的评论下面学到的,太牛啦!!):

        只循环n/2+1次(n是合并后数组的长度),其实就是刚好走到中位数

        设置两个位置:left跟right,right用来往前找

        nums1移动的条件:nums1还没遍历完,且nums1<nums2

        nums2移动的条件:nums2<nums2

这样做的话可以不需要申请额外的空间来存储排序后的数组,空间复杂度减少为O(1),时间复杂度却没咋改善,还是要遍历数组。

代码在此:

class Solution:
    def findMedianSortedArrays(self, A: List[int], B: List[int]) -> float:
        m = len(A)
        n = len(B) 
        lens = m + n
        left, right = -1, -1
        aStart, bStart = 0, 0
        for i in range(lens//2 + 1) :
            left = right  # 每次循环前将 right 的值赋给 left
            # A移动的条件: B遍历到最后 或 当前A<B,满足一个即可
            if aStart < m and (bStart >= n or A[aStart] < B[bStart]):
                right = A[aStart]
                aStart += 1
            else :
                right = B[bStart]
                bStart += 1
            
        if (lens & 1) == 0: # 与1交,判断奇偶数,更快速
            return (left + right) / 2.0
        else:
            return right

比较普遍的想法:找出中位数 = > 找出第k小的数,其实就是不断减去元素,最后就只剩下中位数。先根据两个数组的长度求出中位数所在位置k,即求第k小的数。每次循环排除k/2的数,可以看下面第一个例子,先找到两个数组第k/2的数字,小的那一个前面所有的数就肯定不会是中位数,就是第一个例子里面的【1,2,3】这三个。到第二个例子,就要找第7-3=4小的数了(因为已经排除了三个数)

 

 

 

 代码在此:下回再看吧

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值