leetcode-004-Median_of_Two_Sorted_Arrays

发表于 | 更新于: | 分类于 |

leetcode-004-Median-of-Two-Sorted-Arrays 寻找两个有序数组的中位数

1. 题目

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

2. 解

暴力解法：把两个有序数组排序为一个有序数组，判断数组长度是奇数还是偶数， 然后返回中位数。这种解法的时间复杂度是O(m+n)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        nums = []
        i,j = 0,0
        for k in range(len(nums1) + len(nums2)):
            if (i < len(nums1)) & (j < len(nums2)):
                if nums1[i] < nums2[j]:
                    nums.append(nums1[i])
                    i += 1
                else:
                    nums.append(nums2[j])
                    j += 1
            elif i < len(nums1):
                nums.append(nums1[i])
                i += 1
            elif j < len(nums2):
                nums.append(nums2[j])
                j += 1
        if(len(nums)%2):
            return float(nums[len(nums)//2])
        else:
            return (nums[len(nums)//2 - 1] + nums[len(nums)//2])/float(2)