在 Python 中从算术序列中查找已删除项的程序
假设我们有一个名为nums的数组,其中包含n-1个等差数列项。之前删除了除nums的第一个或最后一个元素之外的一个元素。我们必须找到被删除的号码。
因此,如果输入类似于nums=[5,7,11,13],则输出将为9,因为项目遵循公式2i+5,因此对于i=2,它将是2*2+5=9缺少。
示例
让我们看下面的实现来更好地理解
def solve(nums):
if len(nums) == 2:
return sum(nums) //2
if nums[0] == nums[1]:
return nums[0]
lower = nums[0]
upper = nums[-1]
interval = (upper - lower) //长度(编号)
pointer = len(nums) //2
left = 0
right = len(nums) - 1
while left != right:
if nums[pointer] != nums[0] + interval * pointer:
if nums[pointer - 1] == nums[0] + interval * (pointer -1):
return nums[0] + interval * pointer
else:
right = pointer
pointer = (left + right) //2
else:
if right - left == 1:
pointer = right
else:
left = pointer
pointer = (left + right) //2
nums = [5, 7, 11, 13]
print(solve(nums))输入
[5, 7, 11, 13]输出结果
9