题目:有两个排序的数组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])