def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
n = len(nums1) + len(nums2)
if n % 2 == 1:
return self.findKth(nums1, nums2, int(n / 2) + 1) * 1.0
else:
smaller = self.findKth(nums1, nums2, int(n / 2))
bigger = self.findKth(nums1, nums2, int(n / 2) + 1)
return (smaller + bigger) / 2.0
def findKth(self, A, B, k):
print(A, B, k)
if len(A) == 0:
return B[k - 1]
if len(B) == 0:
return A[k - 1]
if k == 1:
return min(A[0], B[0])
a = A[int(k / 2) - 1] if len(A) >= k / 2 else None
b = B[int(k / 2) - 1] if len(B) >= k / 2 else None
if b is None or (a is not None and a < b):
return self.findKth(A[int(k / 2):], B, k - int(k / 2))
return self.findKth(A, B[int(k / 2):], k - int(k / 2))
Leetcode 之 Median of Two Sorted Arrays
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- https://leetcode.com/problems/median-of-two-sorted-arrays...
- Description There are two sorted arrays nums1 and nums2 o...
- 题目 There are two sorted arrays nums1 and nums2 of size m ...
- There are two sorted arrays nums1 and nums2 of size m and...
- 再让我说一段正经的话,你不远走,我不离去。 311. 你有一秒没说话 你成了谁 我在我的心上,画了好久你的样子 3...