力扣刷题记录-4 寻找两个正序数组的中位数

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        i=0 #i遍历nums1 j遍历nums2
        j=0
        tmp=[]
        if nums1 == []:
            tmp=nums2
        elif nums2 == []:
            tmp=nums1
        else:
            while(i < len(nums1) and j < len(nums2)):
                if nums1[i] < nums2[j]:
                    if i == len(nums1)-1:
                        tmp.append(nums1[i])
                        tmp.extend(nums2[j:])
                        break
                    tmp.append(nums1[i])
                    i=i+1
                if nums1[i] >= nums2[j]:
                    if j == len(nums2)-1:
                        tmp.append(nums2[j])
                        tmp.extend(nums1[i:])
                        break
                    tmp.append(nums2[j])
                    j=j+1
        if len(tmp) % 2 == 1:      #如果是奇数
            result=tmp[int(len(tmp)/2)]
        else:
            result=(tmp[int(len(tmp)/2)]+tmp[int(len(tmp)/2)-1])/2
        return result

思路:合并排序

由于题目给出的两个数组已经有序,所以遍历两个数组,在遍历过程中对比两个数组的元素哪个小,将小的插入到新的数组中,从而得到一个排好序的合并数组

(1)令i指向nums1,j指向nums2

(2)对比i所指向的元素和j所指向的元素哪个小,将较小的元素放入tmp数组中

(3)在将较小元素的放入后,其对应的指针向后走

(4)重复(2)和(3)直到两个数组都遍历完成

(5)算出中位数,要注意长度为奇数还是偶数的区别

优化思路(代码未优化):

根据nums1和nums2的长度可以快速求出计算中位数所需要元素的位置,遍历数组时遍历到该位置即可停止,省去合并成新数组的空间复杂度以及时间复杂度

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值