Python【寻找缺失元素】

要求：

    给定一个非负整数数组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_element

A = [1, 2, 4, 5]
missing = find_missing_element(A)
print(f"缺失的元素为：{missing}")