题目描述:
给定两个大小分别为 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
class Solution(object):
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
self.nums1=nums1
self.nums2=nums2
l1=len(self.nums1)
l2=len(self.nums2)
target=[]
i=0
j=0
while i<l1 and j<l2:
if self.nums1[i]>self.nums2[j]:
target.append(self.nums2[j])
j+=1
elif self.nums1[i]<self.nums2[j]:
target.append(self.nums1[i])
i+=1
elif self.nums1[i]==self.nums2[j]:
target.append(self.nums1[i])
target.append(self.nums2[j])
i+=1
j+=1
if i<l1:
target.extend(self.nums1[i:])
if j<l2:
target.extend(self.nums2[j:])
l3=len(target)
if l3%2==0:
s=float(target[l3/2-1]+target[l3/2])/2.0
return s
else:
return float(target[l3//2])