要求:
给定一个非负整数数组A,该数组的元素都是从1~N的元素组成,现在发现其中缺失一个,请找出这个缺失的元素
代码如下:
def find_missing_element(arr):"""找出缺失的元素:param arr: 包含1~N之间的整数的数组,其中有一个元素缺失:return: 缺失的元素"""n = len(arr) + 1 # N+1个元素,包含了缺失的元素expected_sum = n * (n + 1) // 2 # 1到N的和(等差数列)actual_sum = sum(arr) # 数组中所有元素的和missing_element = expected_sum - actual_sum # 计算缺失的元素return missing_elementA = [1, 2, 4, 5]
missing = find_missing_element(A)
print(f"缺失的元素为:{missing}")