题目:有两个排序的数组nums1和nums2,长度分别为m和n。找到两个排序数组的中位数。总运行时间复杂度应为O(log(m + n))。可以假设nums1和nums2 不能都为空。
例:nums1 = [1,3] nums2 = [2] 中位数是2.0;nums1 = [1,2] nums2 = [3,4] 中位数为(2 + 3)/ 2 = 2.5
思路:可以将两个数组列表组合形成一个新的列表,再对新的列表排序,如果(m+n)%2==0,则中位数为中间两个数的均值,如果(m+n)%2==1,则中位数为最中间一个数。
代码如下:
使用位移操作符 (m+n) >> 1比 int((m+n) /2) 慢
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
m,n=len(nums1),len(nums2)
list_merge=sorted(nums1+nums2)
median_i=int((m+n)/2)
if (m+n)%2==0:
return (list_merge[median_i-1]+list_merge[median_i])/2
else:
return float(list_merge[median_i])
本文介绍了一种算法,用于查找两个已排序数组的中位数,总运行时间复杂度为O(log(m+n))。通过合并并排序两个数组,然后根据数组总长度的奇偶性确定中位数。
787

被折叠的 条评论
为什么被折叠?



