给定两个大小为 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
方法一:内置函数
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
nums = nums1 + nums2
nums.sort()
n = len(nums)
if n % 2 != 0:
return float(nums[n // 2])
return (nums[n // 2] + nums[n // 2 - 1]) / 2
方法二:双指针,通过双指针将两个数组合并成一个新数组,再求中位数。
class Solution:
def findMedianSortedArrays(self, nums1: 'List[int]', nums2: 'List[int]') -> float:
m = len(nums1)
n = len(nums2)
i, j = 0, 0
result = []
while i < m or j < n:
if len(result) > (m + n) // 2:
break
if i < m and j < n:
if nums1[i] > nums2[j]:
result.append(nums2[j])
j += 1
else:
result.append(nums1[i])
i += 1
elif i >= m:
result.append(nums2[j])
j += 1
elif j >= n:
result.append(nums1[i])
i += 1
if (m + n) % 2 == 1:
return result[(m + n) // 2]
else:
return (result[(m + n) // 2] + result[(m + n) // 2 - 1]) / 2