编程练习系列-findMedianSortedArrays实现

最新推荐文章于 2024-08-10 21:27:00 发布

Amelia0911

最新推荐文章于 2024-08-10 21:27:00 发布

阅读量434

点赞数 1

分类专栏： python编程文章标签： python leetcode algorithm

本文链接：https://blog.csdn.net/liuzhuomei0911/article/details/110519777

版权

python编程专栏收录该内容

6 篇文章

订阅专栏

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。
输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

思路：
首先判断其中任意一个数组是否为空或遍历完；
若不为空或没有遍历结束，继续对比进行合并操作；
最后求中位数。

from typing import List

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        len1 = nums1.__len__()
        len2 = nums2.__len__()

        nums = []
        i = j = 0
        while 1:
            if i >= len1:
                nums += nums2[j:]
                break
            if j >= len2:
                nums += nums1[i:]
                break

            if nums1[i] < nums2[j]:
                nums.append(nums1[i])
                i += 1
            else:
                nums.append(nums2[j])
                j += 1

        return (nums[int((len1 + len2) / 2)] if (len1 + len2) % 2 != 0 else (nums[int((len1 + len2) / 2) - 1] + nums[int((len1 + len2) / 2)]) / 2)


if __name__ == '__main__':

     l1 = [1, 2]
     l2 = [3, 4,5]
     myfun = Solution()
     dst = myfun.findMedianSortedArrays(l1, l2)
     print("dst = ", dst)