问题:
我正在完成第 6 版的《Cracking the Coding Interview》,但遇到了 Circus Tower 问题(编号 17.8)。我有一个我认为在 O(N logN) 时间内运行的解决方案,但本书的解决方案(不同)说 O(N logN) 解决方案非常复杂,但我的代码不是。我需要一些帮助来确定我的解决方案是否正确,以及它是否真的在 O(N logN) 时间内运行。我想了解我为什么错(或对),所以任何细节都会有所帮助。
这是问题文本:
一个马戏团正在设计一个由站在彼此肩上的人组成的塔式程序。出于实用和审美的原因,每个人都必须比他或她下面的人更矮更轻。给定马戏团中每个人的身高和体重,编写一个方法来计算这样一个塔中可能存在的最大人数。
我的解决方案:
def circus_tower(people):
# People is a list of tuples of length 2, where item[0] is their height, and item[1] is their weight.
if len(people) == 0:
return 0
people = sorted(people, key=lambda x: x[0]) # Sort people by heights. O(P*logP).
start = 0
end = 0
longest_sequence = 0
for i in range(1, len(people)): # O(P).
if people[i][1] > people[i-1][1]: # Weight check.
end = i
else:
longest_sequence = end - start + 1
start = i
end = i
return max(longest_sequence, end - start + 1)
以下是一些示例输入以及我的代码返回的内容:
circus_tower([(65, 100), (70, 150), (56, 90), (75, 190), (60, 95), (68, 110)])
6
circus_tower([(65, 100), (70, 150), (56, 90), (75, 190), (65, 95), (68, 110)])
4
circus_tower([(56, 90), (65, 95), (72, 100), (68, 90), (70, 150), (75, 190)])
2
circus_tower([])
0