第一道困难题
题目地址
- 寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
解法地址
用了递归思想
找中位数其实就是找第k小的数问题,解法地址里面有详尽解释
对于偶数情况,k1比k2小1,中位数为k1和k2指向的两个数的均值
对于奇数情况,k1和k2相等,中位数为它们所指向的数
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
k1 = ( len(nums1) + len(nums2) + 1 ) // 2
k2 = ( len(nums1) + len(nums2) + 2 ) // 2
def helper(nums1,nums2,k): #本质上是找第k小的数
print('k:',k)
if(len(nums1) <len(nums2) ):
nums1, nums2 = nums2 , nums1 #保持nums1比较长
if(len(nums2)==0):
return nums1[k-1] # 短数组空,直接返回
if(k==1):
return min(nums1[0],nums2[0]) #找最小数,比较数组首位
t = min(k//2,len(nums2)) # 保证不上溢
if( nums1[t-1]>=nums2[t-1] ):
return helper(nums1 , nums2[t:],k-t)
else:
return helper(nums1[t:],nums2,k-t)
return ( helper(nums1,nums2,k1) + helper(nums1,nums2,k2) ) /2
注意return是在helper函数外面的。