两个排序数组的中位数

程序小学生

于 2018-11-17 13:18:54 发布

阅读量143

点赞数

分类专栏： LeetCode 文章标签：两个排序数组的中位数 Python LeetCode

本文链接：https://blog.csdn.net/qq_25174673/article/details/84182301

版权

LeetCode 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

4.两个排序数组的中位数

题目描述

给定两个大小为 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

解题思路

新建一个空列表，把nums1和nums2的两个数组内容按顺序放入空列表中，然后根据nums1和nums2的长度，当res达到取中位数的长度时就跳出，返回结果。

Python代码如下

class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        m = len(nums1)
        n = len(nums2)
        res = []
        i,j = 0,0
        while i<m or j <n:
            if i<m:
                x = nums1[i]
            else:
                x = 999999999
            if j<n:
                y = nums2[j]
            else:
                y = 999999999
            if x <= y:
                res.append(x)
                i +=1
            else:
                res.append(y)
                j +=1
            if len(res) > (m+n)//2:
                break
        if (m+n)%2 == 0:
            a = (m+n)//2
            b = a-1
            return (res[a]+res[b])/2
        else:
            a = (m+n)//2
            return res[a]