题目:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
示例 1:
nums1 = [1, 3]
nums2 = [2]
中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
中位数是 (2 + 3)/2 = 2.5
利用Python list 的 sort 函数, 这道题还是很容易实现,主要是需要判断list 元素个数是奇数还是偶数。Python3 编码参考如下:
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
self.nums = nums1 + nums2
self.nums.sort()
size = len(self.nums)
if size % 2 == 1:
return self.nums[size//2]
if size % 2 == 0:
return (self.nums[size//2] + self.nums[size//2 - 1])/2
ps:如果您有好的建议,欢迎交流 :-D,也欢迎访问我的个人博客:tundrazone.com
网友评论