力扣刷题：寻找两个正序数组的中位数（python3）

题目描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：
输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：
输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

解法一

这种解法按道理来说时间复杂度是不符合的，属于Python自带的赖皮写法，没有太大参考意义。

代码

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        new_nums = nums1+nums2
        new_nums.sort()
        return (new_nums[len(new_nums)//2]+new_nums[(len(new_nums)-1)//2])/2

解法二

这种解法不仅适用于求中位数，也可以用来求最小k位数、最大k位数。

代码

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        add_lenth = len(nums1)+len(nums2)
        i,j,count,new_list = 0,0,0,[]
        if nums1 and nums2:
            while count <= ((add_lenth //2)):
                    if i>=len(nums1):
                        new_list.append(nums2[j])
                        j += 1
                        count += 1
                    elif j>=len(nums2):
                        new_list.append(nums1[i])
                        i += 1
                        count += 1
                    else:
                        if nums1[i] <= nums2[j]:
                            new_list.append(nums1[i])
                            i += 1
                            count += 1
                        else:
                            new_list.append(nums2[j])
                            j += 1
                            count += 1
            return (new_list[add_lenth//2]+new_list[(add_lenth-1)//2])/2
        elif nums1:
            return (nums1[len(nums1)//2]+nums1[(len(nums1)-1)//2])/2
            
        elif nums2:
            return (nums2[len(nums2)//2]+nums2[(len(nums2)-1)//2])/2

具体讲解可以参考这个视频
https://leetcode.cn/problems/median-of-two-sorted-arrays/solutions/210764/di-k-xiao-shu-jie-fa-ni-zhen-de-dong-ma-by-geek-8m/