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

题目

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

示例 1：

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

思路

首先想到的还是笨办法，将 nums1 和 nums2 按升序合并在一起，存储在nums3中，然后求 muns3 的中位数

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:

        if len(nums1)==0 and len(nums2)==0:
            median=None

        if len(nums1)==0 and len(nums2)!=0:
            if len(nums2)%2==0:
                median=(nums2[int(len(nums2)/2)]+nums2[int(len(nums2)/2-1)])/2
            else:
                median= nums2[int((len(nums2)-1)/2)]

        if len(nums2)==0 and len(nums1)!=0:
            if len(nums1)%2==0:
                median=(nums1[int(len(nums1)/2)]+nums1[int(len(nums1)/2-1)])/2
            else:
                median= nums1[int((len(nums1)-1)/2)]

        global nums3
        nums3=[]      # nums3 存储 nums1和nums2合并以后的数组
        if len(nums1)!=0 and len(nums2)!=0:
            i=0
            j=0
            while i<len(nums1) and j<len(nums2):
                if nums1[i]<=nums2[j]:
                    nums3.append(nums1[i])
                    i+=1
                else:
                    nums3.append(nums2[j])
                    j+=1
            if i<len(nums1) and j>=len(nums2):
                while i<len(nums1):
                    nums3.append(nums1[i])
                    i+=1
            if i>=len(nums1) and j<len(nums2):
                while j<len(nums2):
                    nums3.append(nums2[j])
                    j+=1
            
            if len(nums3)%2==0:
                median=(nums3[int(len(nums3)/2)]+nums3[int(len(nums3)/2-1)])/2  
            else:
                median=nums3[int((len(nums3)-1)/2)]

        return median

方法2

使用归并排序的方法

#  归并排序的递归方法
import math

def mergeSort(aa):
    if len(aa)<2:
        return aa
    middle = math.floor(len(aa)/2)
    left, right = aa[0:middle], aa[middle:]
    return sort(mergeSort(left), mergeSort(right))


def sort(bb, cc):
    result = []
    while bb and cc:
        if bb[0] <= cc[0]:
            result.append(bb.pop(0))
        else:
            result.append(cc.pop(0))
    while bb:
        result.append(bb.pop(0))
    while cc:
        result.append(cc.pop(0))

    return result

aa=[4,2,1,5,8,9,7,4]
print(mergeSort(aa))