给定两个大小为 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)
暂时阵亡
等我想到其他思路再战!