lg n 算法是将输入分成更小的部分,并丢弃一些更小的部分,这样您就可以使用更小的输入。由于这是一个搜索问题,因此 lg n 时间复杂度的可能解决方案是二分搜索,每次将输入分成两半。
我的方法是从几个简单的案例开始,找出我可以利用的任何模式。
在以下示例中,最大整数是目标数。
# input size: 3
[1,1,2]
[2,1,1]
# input size: 5
[1,1,2,2,3]
[1,1,3,2,2]
[3,1,1,2,2]
# input size: 7
[1,1,2,2,3,3,4]
[1,1,2,2,4,3,3]
[1,1,4,2,2,3,3]
[4,1,1,2,2,3,3]
# input size: 9
[1,1,2,2,3,3,4,4,5]
[1,1,2,2,3,3,5,4,4]
[1,1,2,2,5,3,3,4,4]
[1,1,5,2,2,3,3,4,4]
[5,1,1,2,2,3,3,4,4]
您可能注意到输入大小始终是奇数,即2*x + 1。
由于这是一个二分搜索,您可以检查中间的数字是否是您的目标数字。如果中间的数字是单个数字(if middle_number != left_number and middle_number != right_number),那么你已经找到了。否则,您必须搜索输入的左侧或右侧。
请注意,在上面的示例测试用例中,中间数字不是目标数字,中间数字与其对之间存在模式。
对于输入大小3(2*1 + 1),if middle_number == left_number,目标数在右边,反之亦然。
对于输入大小 5 (2*2 + 1),if middle_number == left_number,目标数字在左边,反之亦然。
对于输入大小 7 (2*3 + 1),if middle_number == left_number,目标数在右边,反之亦然。
对于输入大小 9 (2*4 + 1),if middle_number == left_number,目标数在左边,反之亦然。
这意味着2*x + 1中x的奇偶性(数组长度)影响是搜索输入的左边还是右边:如果x是奇数则搜索右边,如果x是偶数则搜索左边,如果middle_number = = left_number(反之亦然)。
基于所有这些信息,您可以提出递归解决方案。请注意,您必须确保每个递归调用中的输入大小都是奇数。 (编辑:确保输入大小是奇数会使代码更加混乱。您可能想提出一个解决方案,其中输入大小的奇偶性无关紧要。)
def find_single_number(array: list, start_index: int, end_index: int):
# base case: array length == 1
if start_index == end_index:
return start_index
middle_index = (start_index + end_index) // 2
# base case: found target
if array[middle_index] != array[middle_index - 1] and array[middle_index] != array[middle_index + 1]:
return middle_index
# make use of parity of array length to search left or right side
# end_index == array length - 1
x = (end_index - start_index) // 2
# ensure array length is odd
include_middle = (middle_index % 2 == 0)
if array[middle_index] == array[middle_index - 1]: # middle == number on its left
if x % 2 == 0: # x is even
# search left side
return find_single_number(
array,
start_index,
middle_index if include_middle else middle_index - 1
)
else: # x is odd
# search right side side
return find_single_number(
array,
middle_index if include_middle else middle_index + 1,
end_index,
)
else: # middle == number on its right
if x % 2 == 0: # x is even
# search right side side
return find_single_number(
array,
middle_index if include_middle else middle_index + 1,
end_index,
)
else: # x is odd
# search left side
return find_single_number(
array,
start_index,
middle_index if include_middle else middle_index - 1
)
# test out the code
if __name__ == '__main__':
array = [2,2,1,1,3,3,4,5,5,6,6] # target: 4 (index: 6)
print(find_single_number(array, 0, len(array) - 1))
array = [1,1,2] # target: 2 (index: 2)
print(find_single_number(array, 0, len(array) - 1))
array = [1,1,3,2,2] # target: 3 (index: 2)
print(find_single_number(array, 0, len(array) - 1))
array = [1,1,4,2,2,3,3] # target: 4 (index: 2)
print(find_single_number(array, 0, len(array) - 1))
array = [5,1,1,2,2,3,3,4,4] # target: 5 (index:0)
print(find_single_number(array, 0, len(array) - 1))
我的解决方案可能不是最有效或最优雅的,但我希望我的解释能帮助您理解解决这类算法问题的方法。
证明它的时间复杂度为 O(lg n):
假设最重要的操作是中间数与左右数(if array[middle_index] != array[middle_index - 1] and array[middle_index] != array[middle_index + 1])的比较,时间成本为 1 个单位。让我们将此比较称为主要比较。
令 T 为算法的时间成本。
设 n 为数组的长度。
由于此解决方案涉及递归,因此存在基本情况和递归情况。
对于基本情况(n = 1),它只是主要的比较,所以:
T(1) = 1。
对于递归情况,每次将输入分成两半(左半部分或右半部分);同时,还有一个主要的比较。所以:
T(n) = T(n/2) + 1
现在,我知道输入大小必须总是奇数,但为了简单起见,我们假设 n = 2k;时间复杂度还是一样的。
我们可以将 T(n) = T(n/2) + 1 重写为:
T(2k) = T(2k-1) + 1
另外,T(1) = 1 是:
T(20) = 1
当我们展开 T(2k) = T(2k-1) + 1 时,我们得到:
T(2k)
= T(2k-1) + 1
= [T(2k-2) + 1] + 1 = T(2k-2) + 2
= [T(2k-3) + 1] + 2 = T(2k-3) + 3
= [T(2k-4) + 1] + 3 = T(2k-4) + 4
= ...(重复直到 k)
= T(2k-k) + k = T(20) + k = k + 1
由于n = 2k,这意味着k = log2 n.
将 n 代入,我们得到:
T(n) = log2 n + 1
1 是一个常数,所以它可以被删除;日志操作的基础也是如此。
因此,算法时间复杂度的上界为:
T(n) = lg n