LeetCode 4. 寻找两个正序数组的中位数
Apale 7/7/2020
累了……以后只放一份代码了。
思路
假设第一个数组的长度为
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
inf = 2000000000
if len(nums1) > len(nums2):
nums1, nums2 = nums2, nums1
s1, s2 = len(nums1), len(nums2)
low, high = 0, 2 * len(nums1)
while low <= high:
m1 = low + high >> 1
m2 = s1 + s2 - m1
l1 = nums1[m1 - 1 >> 1] if m1 else -inf
r1 = nums1[m1 >> 1] if m1 < 2 * s1 else inf
l2 = nums2[m2 - 1 >> 1] if m2 else -inf
r2 = nums2[m2 >> 1] if m2 < 2 * s2 else inf
if l1 > r2:
high = m1 - 1
elif l2 > r1:
low = m1 + 1
else:
break
return (max(l1, l2) + min(r1, r2)) / 2.0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
