【发布时间】:2019-08-11 12:10:08
【问题描述】:
我在https://www.geeksforgeeks.org/median-of-two-sorted-arrays/ 上找到了计算 2 个排序列表中位数的算法。 它说,它是 O(log(n))。 但真的是这样吗?
我感到困惑的是: 这些行将一个数组拆分为 2 个子数组(使用 Python 的切片)并递归求解:
if n % 2 == 0:
return getMedian(arr1[:int(n / 2) + 1],
arr2[int(n / 2) - 1:], int(n / 2) + 1)
else:
return getMedian(arr1[:int(n / 2) + 1],
arr2[int(n / 2):], int(n / 2) + 1)
但对我来说,拆分数组看起来像 O(n)。 所以在我看来,整个算法一定是O(n * log n)...
在这里,您可以看到我正在谈论的算法的整个代码:
# using divide and conquer we divide
# the 2 arrays accordingly recursively
# till we get two elements in each
# array, hence then we calculate median
#condition len(arr1)=len(arr2)=n
def getMedian(arr1, arr2, n):
# there is no element in any array
if n == 0:
return -1
# 1 element in each => median of
# sorted arr made of two arrays will
elif n == 1:
# be sum of both elements by 2
return (arr1[0]+arr2[1])/2
# Eg. [1,4] , [6,10] => [1, 4, 6, 10]
# median = (6+4)/2
elif n == 2:
# which implies median = (max(arr1[0],
# arr2[0])+min(arr1[1],arr2[1]))/2
return (max(arr1[0], arr2[0]) +
min(arr1[1], arr2[1])) / 2
else:
#calculating medians
m1 = median(arr1, n)
m2 = median(arr2, n)
# then the elements at median
# position must be between the
# greater median and the first
# element of respective array and
# between the other median and
# the last element in its respective array.
if m1 > m2:
if n % 2 == 0:
return getMedian(arr1[:int(n / 2) + 1],
arr2[int(n / 2) - 1:], int(n / 2) + 1)
else:
return getMedian(arr1[:int(n / 2) + 1],
arr2[int(n / 2):], int(n / 2) + 1)
else:
if n % 2 == 0:
return getMedian(arr1[int(n / 2 - 1):],
arr2[:int(n / 2 + 1)], int(n / 2) + 1)
else:
return getMedian(arr1[int(n / 2):],
arr2[0:int(n / 2) + 1], int(n / 2) + 1)
# function to find median of array
def median(arr, n):
if n % 2 == 0:
return (arr[int(n / 2)] +
arr[int(n / 2) - 1]) / 2
else:
return arr[int(n/2)]
# Driver code
arr1 = [1, 2, 3, 6]
arr2 = [4, 6, 8, 10]
n = len(arr1)
print(int(getMedian(arr1,arr2,n)))
# This code is contributed by
# baby_gog9800
【问题讨论】:
-
在 Python 中,如果你对一个列表进行切片,你会创建一个副本,因此这将花费 O(n)。 GFG 文章中的切片不会创建副本,它只是保存列表如何切片的指针。
-
它将两个数组分成两半,所以从 2n 大小你去 n/2+n/2=n 所以你把问题减半了。虽然如果它只是一个大小为 n 的数组,那么执行 n/2+n/2=n 仍然是 O(n)。
标签: algorithm runtime time-complexity