1、题目:找出两个有序数组的中位数!
例子:(就这么简单粗暴嘞)
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
2、解题思路:
常规想法哈:最暴力的解法就是先拼接两个数组,再根据定义去找出合并后的数组的中位数。但是这个方法实在是太“暴力”了,时间复杂度跟空间度都很大。在这个基础之上进行优化(我在力扣这题的评论下面学到的,太牛啦!!):
只循环n/2+1次(n是合并后数组的长度),其实就是刚好走到中位数
设置两个位置:left跟right,right用来往前找
nums1移动的条件:nums1还没遍历完,且nums1<nums2
nums2移动的条件:nums2<nums2
这样做的话可以不需要申请额外的空间来存储排序后的数组,空间复杂度减少为O(1),时间复杂度却没咋改善,还是要遍历数组。
代码在此:
class Solution:
def findMedianSortedArrays(self, A: List[int], B: List[int]) -> float:
m = len(A)
n = len(B)
lens = m + n
left, right = -1, -1
aStart, bStart = 0, 0
for i in range(lens//2 + 1) :
left = right # 每次循环前将 right 的值赋给 left
# A移动的条件: B遍历到最后 或 当前A<B,满足一个即可
if aStart < m and (bStart >= n or A[aStart] < B[bStart]):
right = A[aStart]
aStart += 1
else :
right = B[bStart]
bStart += 1
if (lens & 1) == 0: # 与1交,判断奇偶数,更快速
return (left + right) / 2.0
else:
return right
比较普遍的想法:找出中位数 = > 找出第k小的数,其实就是不断减去元素,最后就只剩下中位数。先根据两个数组的长度求出中位数所在位置k,即求第k小的数。每次循环排除k/2的数,可以看下面第一个例子,先找到两个数组第k/2的数字,小的那一个前面所有的数就肯定不会是中位数,就是第一个例子里面的【1,2,3】这三个。到第二个例子,就要找第7-3=4小的数了(因为已经排除了三个数)
代码在此:下回再看吧