题目描述
给定两个大小为 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(object):
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
m = len(nums1)
n = len(nums2)
mid = 0.0
for i in range (n):
nums1.append(nums2[i])
nums1.sort()
a = m+n
if a%2 == 0:
# print(nums1[a//2 -1] + nums1[a//2])
mid = (nums1[a//2 -1] + nums1[a//2]) * 0.5
# print(mid)
else:
mid = nums1[a//2]
return mid