leetcode-4-寻找两个有序数组的中位数

发表于 | 更新于 | 分类于 | 评论数： | 阅读次数：

题目链接

寻找两个有序数组的中位数

题目描述

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

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

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

示例 1:

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

则中位数是 2.0

示例 2:

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

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

解答

思路是对两个数组排序后求中位数。

方法一

使用sorted排序，取中位数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def twoSum(nums, target):
	n=len(nums)
	for i in range(n):class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        nums = sorted(nums1 + nums2)
        l = len(nums)
        if l % 2 == 0:
            return (nums[l // 2 - 1] + nums[l // 2]) / 2
        else:
            return nums[l // 2]

方法二

自己实现排序，然后取中位数。

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
28
29
30
31
32
class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        m = len(nums1)
        n = len(nums2)
        sort_nums = []
        i = 0
        j = 0
        l = m + n
        now_index = 0
        while i < m and j < n:
            if nums1[i] < nums2[j]:
                sort_nums.append(nums1[i])
                i += 1
                now_index += 1
            else:
                sort_nums.append(nums2[j])
                j += 1
                now_index += 1
        if i < m:
            sort_nums.extend(nums1[i:])
        if j < n:
            sort_nums.extend(nums2[j:])

        if l % 2 == 0:
            print((sort_nums[l // 2 - 1] + sort_nums[l // 2]) / 2)
        else:
            print(sort_nums[l // 2])