【Leetcode-python】4.Median of Two Sorted Arrays(两个排序数组的中位数)

题目:有两个排序的数组nums1和nums2,长度分别为m和n。找到两个排序数组的中位数。总运行时间复杂度应为O(log(m + n))。可以假设nums1和nums2  不能都为空。
例:nums1 = [1,3] nums2 = [2]  中位数是2.0;nums1 = [1,2]  nums2 = [3,4]  中位数为(2 + 3)/ 2 = 2.5

思路:可以将两个数组列表组合形成一个新的列表,再对新的列表排序,如果(m+n)%2==0,则中位数为中间两个数的均值,如果(m+n)%2==1,则中位数为最中间一个数。

代码如下:
使用位移操作符 (m+n) >> 1比 int((m+n) /2) 慢

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m,n=len(nums1),len(nums2)
        list_merge=sorted(nums1+nums2)
        median_i=int((m+n)/2)
        if (m+n)%2==0:
             return (list_merge[median_i-1]+list_merge[median_i])/2
        else:
             return float(list_merge[median_i])

你可能感兴趣的:(算法)