leetcode4 python 两个排序数组的中位数

最新推荐文章于 2024-07-12 23:40:30 发布

原创最新推荐文章于 2024-07-12 23:40:30 发布 · 156 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

52 篇文章

订阅专栏

本文探讨了如何在两个已排序的数组中找到中位数，提出了一个算法，但其时间复杂度超过O(log(m+n))的要求，未能达到最优解。示例展示了不同情况下的中位数计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本答案时间复杂度超了，未提交成功

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

你可以假设 nums1 和 nums2 不同时为空。

示例 1:

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

中位数是 2.0
示例 2:

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

中位数是 (2 + 3)/2 = 2.5

class Solution:
    def findMedianSortedArrays(self, num1, num2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        i=0
        j=0
        sort_list = []
        minL=min(len(num1)-1,len(num2)-1)
        if len(num1) == 0:
            if len(num2) % 2 == 1:
                return num2[int(len(num2)/2)]
            else:
                return (num2[int(len(num2)/2)] + num2[int(len(num2)/2 -1)])/2

        if len(num2) == 0:
            if len(num1) % 2 == 1:
                return num1[int(len(num1)/2)]
            else:
                return (num1[int(len(num1)/2)] + num1[int(len(num1)/2 -1)])/2
        while i < minL and j<minL:
            while num1[i] < num2[j] and i < minL:
                sort_list.append(num1[i])
                i+=1
            while num1[i]>num2[j] and j< minL:
                sort_list.append(num2[j])
                j+=1
        sort_list += num1[i:]
        sort_list += num2[j:]
        l=len(num1)+len(num2)
        if l % 2 ==0:
            return (sort_list[int(l/2)]+sort_list[int(l/2)-1])/2
        else:
            return sort_list[math.ceil(l/2)]