題目來自leetcode
https://oj.leetcode.com/problems/median-of-two-sorted-arrays/
There are two sorted arrays A and B 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)).
class Solution:
# @return a float
def findMedianSortedArrays(self, A, B):
lenB = len(B)
lenA = len(A)
if (lenA > lenB):
return self.findMedianSortedArrays(B, A)
#中位點
k = int((lenA + lenB - 1)/2)
#左端
l = int(0)
#右端
r = min(k, lenA)
while l < r:
midA = int((l + r)/2)
midB = int(k - midA)
if (A[midA] < B[midB]):
l = midA + 1
else:
r = midA
choose1 = max(A[l-1], B[k - l]) if (l > 0) else B[k - l]
if (lenA + lenB)&1 == 1:
return choose1
if (lenA <= l):
choose2 = B[k - l + 1]
elif (k - l + 1) >= lenB:
choose2 = A[l]
else:
choose2 = min(A[l], B[k - l + 1])
return float(choose1 + choose2)/2
說明:
這一題的難度是Hard。我花了不少時間。最終的解決思路是通過二分法。
這裏是判斷長度。爲了編程的複雜度,總是使A爲短的一方。(因爲後續代碼操作的基準是A)
if (lenA > lenB):
return self.findMedianSortedArrays(B, A)
這裏k選取兩個list長度之和的中間位置。因爲B爲較長的list,所以後續B從此處開始比較。
l,r是A的兩個比較端點。
#中位點
k = int((lenA + lenB - 1)/2)
#左端
l = int(0)
#右端
r = min(k, lenA)
下面開始比較,基本是二分法的思路,根據比較結果分別移動l和r。應該比較簡單了。
while l < r:
midA = int((l + r)/2)
midB = int(k - midA)
if (A[midA] < B[midB]):
l = midA + 1
else:
r = midA
最後,當循環結束,我們得到以A[l] 和B[k - l]爲邊界的兩組數據(左邊的小,右邊的大)。
如果兩個數組總長度爲奇數,選擇choose1;否則選擇choose1和choose2 平均的結果。
choose1 = max(A[l-1], B[k - l]) if (l > 0) else B[k - l]
if (lenA + lenB)&1 == 1:
return choose1
if (lenA <= l):
choose2 = B[k - l + 1]
elif (k - l + 1) >= lenB:
choose2 = A[l]
else:
choose2 = min(A[l], B[k - l + 1])
return float(choose1 + choose2)/2