-
题目:给定两个大小分别为
m
和n
的正序(从小到大)数组nums1
和nums2
。请你找出并返回这两个正序数组的 中位数 。算法的时间复杂度应该为O(log (m+n))
。1 -
实例:
# 示例一
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
# 示例二
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
- 提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
- 思路:
暴力解法,首先按照顺序依次组合成一个从小到大的列表,然后根据列表长度的奇偶做处理,若为偶数则输出当前取整的中间数与前一个值的平均值;若为奇数,则输出当前取整中间数的前一个数。
- 解法一:
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
lists = []
index_1, index_2 = 0, 0
while index_1 < len(nums1) and index_2 < len(nums2):
if nums1[index_1] < nums2[index_2]:
lists.append(nums1[index_1])
index_1 += 1
elif nums1[index_1] > nums2[index_2]:
lists.append(nums2[index_2])
index_2 += 1
elif nums1[index_1] == nums2[index_2]:
lists.append(nums1[index_1])
index_1 += 1
lists.append(nums2[index_2])
index_2 += 1
while index_1 != len(nums1):
lists.append(nums1[index_1])
index_1 += 1
while index_2 != len(nums2):
lists.append(nums2[index_2])
index_2 += 1
tar = len(lists) // 2
if len(lists) % 2 == 0:
return float((lists[tar-1] + lists[tar])/2)
else:
return float(lists[tar])
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/median-of-two-sorted-arrays ↩︎