1. 程式人生 > >leetcode 4 兩個排序陣列的中位數 python

leetcode 4 兩個排序陣列的中位數 python

給定兩個大小為 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