跳转至

LeetCode: 1228. 等差数列中缺失的数字

1、题目描述

有一个数组,其中的值符合等差数列的数值规律,也就是说:

0 <= i < arr.length - 1 的前提下,arr[i+1] - arr[i] 的值都相等。 我们会从该数组中删除一个 既不是第一个 也 不是最后一个的值,得到一个新的数组 arr

给你这个缺值的数组 arr,请你帮忙找出被删除的那个数。

示例 1:

输入:arr = [5,7,11,13]
输出:9
解释:原来的数组是 [5,7,9,11,13]。

示例 2:

输入:arr = [15,13,12]
输出:14
解释:原来的数组是 [15,14,13,12]。

提示:

  • 3 <= arr.length <= 1000
  • 0 <= arr[i] <= 10^5

2、解题思路

  • 只需要找出两个不同的差值即可判断,缺失的差是两倍
class Solution:
    def missingNumber(self, arr: List[int]) -> int:
        length = len(arr)
        pre = [arr[1] - arr[0], 1]
        ans = 0
        for i in range(2, length):
            cur_diff = arr[i] - arr[i - 1]
            if cur_diff != pre[0]:
                if cur_diff == pre[0] * 2:
                    ans = arr[i] - cur_diff // 2
                else:
                    ans = arr[1] - pre[0] // 2
        return ans