寻找两个有序数组的中位数(Python实现)

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0

示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays

思路一、将两个数组合并为一个数组,重新排序后找到中位数
在这里插入图片描述
从图中可以看出,排序算法最小的时间复杂度仍为O(nlogn),不满足题意

思路二、由于是两个已排好序的数组,可以反复比较两个列表,提取最小的数字组成新的列表,当新列表中元素个数等于m+n/2时,即得到了中位数。但时间复杂度为O(m+n)

思路三、
要想实现时间复杂度为O(log(m+n)),需采用分治算法
一开始使用递归,修改到怀疑人生。应该是从思路上就错了,禁不起大量试验。
错误思路是:分别计算num1和num2的中位数,同时删除同样个数的较大或较小数(分别位于两个数组中)
错误代码如下:

import math


class Solution:
    
    def get_average(self,num):
        index=math.floor(len(num)/2)
        if len(num)==1:
            return num[0]
        if len(num)%2==0:
            
            n=(num[index-1]+num[index])/2
        else:
            n=num[index]
        return n

    def findMedianSortedArrays(self,num1,num2):
        if len(num1)==0 or len(num2)==0:
            if len(num1)==0:
                return self.get_average(num2)
            else:
                return self.get_average(num1)
            
        if num1[0]>num2[0]and num1[-1]<num2[-1]:
            num2=num2[1:]
            num2=num2[:-1]
            return self.findMedianSortedArrays(num1,num2)
        if num2[0]>num1[0]and num2[-1]<num1[-1]:
            num1=num1[1:]
            num1=num1[:-1]
            return self.findMedianSortedArrays(num1,num2)
        
        n1=self.get_average(num1) 
        n2=self.get_average(num2) 
        
        if n1==n2:
            return n1
        if n2<n1:
            num=num1
            num1=num2
            num2=num
            n=n1
            n1=n2
            n2=n
            
        if n1<n2:
            if len(num1)<len(num2):
                if len(num1)==1:
                    num1=[]
                    th=1
                else:
                    th=math.floor(len(num1)/2)
                    
                    num1=num1[th:]
                n=len(num2)-th 
                num2=num2[:n]
            elif len(num1)>len(num2):
                if len(num2)==1:  
                    num2=[]
                    n=1
                    
                else:
                    th=math.ceil(len(num2)/2) 
                    n=len(num2)-th
                    num2=num2[:th]            
                num1=num1[n:]
                
            else:
                if len(num1)==1:
                    return self.get_average(num1+num2)
                else:
                    th=math.ceil(len(num2)/2) 
                    n=len(num2)-th
                    num2=num2[:th]            
                num1=num1[n:]
            return self.findMedianSortedArrays(num1,num2)

暂时阵亡
等我想到其他思路再战!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值