Descripition:
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
解法1:利用归并排序排序思想(Accepted)
由于楼主比较笨,所以第一时间能想到的方法就是利用归并排序,虽然时间复杂度没有做到
O(log(m+n))
O
(
l
o
g
(
m
+
n
)
)
,但是依然 AC 了。这种想法很好实现,只不过需要注意的是边界和奇偶的处理。程序中,我并没有真正去建立一个大数组存储归并之后的所有元素,只是使用 pnums1 和 pnums2 分别记录两个数组归并到了哪里,使用 p 记录已经归并的元素的个数,low 在两数组和为奇数时就是中位数,在两数组和为偶数时是低中位数,这时继续寻找下一个即将归并到大数组中的元素作为高中位数,然后取平均。
class Solution(object):
def findMedianSortedArrays(self, nums1, nums2):
n = len(nums1) + len(nums2)
p = n /2 if n % 2 == 0 else (n + 1)/2
low = 0
pnums1 = pnums2 = 0
while p != 0:
if pnums1 == len(nums1):
low = nums2[pnums2]
pnums2 += 1
elif pnums2 == len(nums2):
low = nums1[pnums1]
pnums1 += 1
elif nums1[pnums1] > nums2[pnums2]:
low = nums2[pnums2]
pnums2 += 1
else:
low = nums1[pnums1]
pnums1 += 1
p -= 1
if n % 2 == 0:
if pnums1 == len(nums1):
high = nums2[pnums2]
elif pnums2 == len(nums2):
high = nums1[pnums1]
else:
high = nums1[pnums1] if nums1[pnums1] < nums2[pnums2] else nums2[pnums2]
return float(low + high) / 2
else:
return low
解法2:二分法(Accepted)
在此题的 Solution 部分已经将这种方法解释的很明白了,并且有 Java 和 Python 的实现,因此这里我就不再说了。