根据题解里面的一个思路写的,啊 感觉 困难 等级的题,好难做==,代码也好难写
有许多需要注意的点:最精妙的就是 利用二分法 找两数组的中间位置:
m1 = left + (right - left) // 2
m2 = k - m1
if nums1[m1] < nums2[m2 - 1]:
left = m1 + 1
else:
right = m1
还有注意下标,从[0, left - 1]是左半部分,left 向右是右半部分
还有判断条件:
m1 > 0 m2 > 0
m1 < n1 m2 < n2
交了第一次wa了,把最后(c1 + c2 )/ 2改成(c1 + c2)*0.5就ac了==
完整代码如下:
class Solution(object):
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
n1 = len(nums1)
n2 = len(nums2)
if n1 > n2:
return self.findMedianSortedArrays(nums2, nums1)
left = 0
right = n1
k = (n1 + n2 + 1) // 2
while left < right:
m1 = left + (right - left) // 2
m2 = k - m1
if nums1[m1] < nums2[m2 - 1]:
left = m1 + 1
else:
right = m1
m1 = left
m2 = k - m1
c1 = max(nums1[m1 - 1] if m1 > 0 else float('-inf'), nums2[m2 - 1] if m2 > 0 else float('-inf'))
if (n1 + n2) % 2 == 1:
return c1
c2 = min(nums1[m1] if m1 < n1 else float('inf'), nums2[m2] if m2 < n2 else float('inf'))
return (c1 + c2) * 0.5