介绍
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。
进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?
示例
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
输入:nums1 = [], nums2 = [1]
输出:1.00000
输入:nums1 = [2], nums2 = []
输出:2.00000
提示
- nums1.length == m
- nums2.length == n
- 0 <= m <= 1000
- 0 <= n <= 1000
- 1 <= m + n <= 2000
- -106 <= nums1[i], nums2[i] <= 106
答案
import copy
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
k1 = int((len(nums1)+len(nums2)+1)/2)
k2 = int((len(nums1)+len(nums2)+2)/2)
def search(a,b,k):
if len(a) > len(b):
a, b = copy.copy(b), copy.copy(a)
if len(a) == 0:
return b[k-1]
if k == 1:
return min(a[0],b[0])
t = int(min(k//2, len(a)))
if a[t-1] < b[t-1]:
return search(a[t:],b,k-t)
else:
return search(a,b[t:],k-t)
return float(search(nums1,nums2,k1)+search(nums1,nums2,k2))/2