4 - Median of Two Sorted Arrays - python

58 阅读1分钟

4. Median of Two Sorted Arrays

中位数,又称中点数,中值。中位数是按顺序排列的一组数据中居于中间位置的数,即在这组数据中,有一半的数据比他大,有一半的数据比他小,

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

You may assume nums1 and nums2 cannot be both empty.

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

AC code

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