题目:
给定两个大小为 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
思路:
1、使用归并的方法将两个有序数组合并为一个大的有序数组,其中间位置即为中位数,时间复杂度为O(m+n),空间复杂度为O(m+n)
2、不需要合并两个有序数组,只要找到中位数的位置即可。维护两个指针,初始时分别指向两个数组的下标 0 的位置,每次将指向较小值的指针后移一位(如果一个指针已经到达数组末尾,则只需要移动另一个数组的指针),直到到达中位数的位置。,时间复杂度为O(m+n),空间复杂度为O(1)
3、由于题目要求时间复杂度为O(log(m+n)),所以想到采用二分的方法,首先分别找到两数组的中间位置,将其分为两个堆,继续调整使得左堆始终小于右堆,找出左堆最大和右堆最小,然后再在两堆的边界返回答案。
代码实现:
from typing import List
class Solution:
# 思想就是利用总