leetcode算法练习 —— 求两个排序排序数组的中位数

Lestat.Z.

于 2019-10-15 21:34:34 发布

阅读量345

点赞数

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.csdn.net/yolohohohoho/article/details/102575033

Python 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种在O(log(min(n,m)))时间内找到两个已排序数组中位数的高效算法。通过将所有元素重组为有序的两部分，中位数位于段点处。文章详细解释了算法的实现步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

给定两个排序过的数组，求这两个排序数组的中位数，在O(log(min(n, m)))内。

解题思路

将所有元素重组成有序的两部分，中位数就在段点处

def find_median(a, b):
	#if len(a) > len(b):
	#	a, b = b, a
	imin, imax = 0, len(a)
	n, m = len(a), len(b)
	while imin <= imax :
		i =  ( imin + imax ) // 2
		j =  ( n + m  + 1) // 2 - i
		if  i < n  and j > 0 and b[j - 1] > a[i]:
			#i is too small
			imin = i + 1
		elif i > 0 and j < m and b[j] < a[i-1]:
			# i is too big
			imax = i - 1
		else:
			# desired case
			if i == 0:
				return b[j-1]
			elif j ==0:
				return a[i-1]
			else:
				return max(a[i-1], b[j-1])