1. 程式人生 > >LeetCode4. 兩個排序陣列的中位數 Python3

LeetCode4. 兩個排序陣列的中位數 Python3

給定兩個大小為 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是常數,那就是這樣子寫了。。。