There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
You may assume nums1 and nums2 cannot be both empty.
Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
解决方法1:最直接的想法,将两个数组合并后排序,再取中位数
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
nums1.extend(nums2)
tmp=sorted(nums1)
if len(tmp)%2==0:
mid=len(tmp)//2
return (tmp[mid]+tmp[mid-1])/2
else:
mid=len(tmp)//2
return tmp[mid]