LeetCode4. 兩個排序陣列的中位數 Python3
阿新 • • 發佈:2018-12-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:
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
num3 = nums1 + nums2
num3.sort()
a = len(num3)/2
if num3 != []:
if a == int(a):
return float((num3[int(a)]+num3[int(a)-1])/2)
else:
return num3[int(a)]
emmm 挺簡單的 O(m+n)的話要是m,n是常數,那就是這樣子寫了。。。