leetcode 4 兩個排序陣列的中位數 python
阿新 • • 發佈:2019-01-09
給定兩個大小為 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 """ tmp = nums1 + nums2 tmp.sort() print tmp if len(tmp)%2==1: return tmp[int(len(tmp)/2)] else: return (tmp[int(len(tmp)/2)-1]+tmp[int(len(tmp)/2)])/2.0