Python, LeetCode, 4. 两个排序数组的中位数

最新推荐文章于 2021-02-03 03:29:18 发布

CherryCheekAir

最新推荐文章于 2021-02-03 03:29:18 发布

阅读量482

点赞数

分类专栏： LeetCode 文章标签： Python LeetCode

本文链接：https://blog.csdn.net/u010342040/article/details/80620012

版权

LeetCode 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        nums = []
        while len(nums1) != 0 and len(nums2) != 0:
            if nums1[0] < nums2[0]:
                nums.append(nums1.pop(0))
            else:
                nums.append(nums2.pop(0))
        if len(nums1) != 0:
            nums += nums1
        else:
            nums += nums2
        if len(nums) % 2 == 0:
            tmp = int(len(nums)/2)
            res = (nums[tmp] + nums[tmp-1])/2
        else:
            res = nums[len(nums)//2]
        return float(res)