Table of Content
题目:
给定两个大小为 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 编码参考如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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 |