LeetCode-Python-1228. 等差数列中缺失的数字（数组）

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

在 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/missing-number-in-arithmetic-progression
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

先找到整个等差数列的差 d，

怎么找？

题目说了第一个数和最后一个数必定有效，

而且等差数列里数的个数为 len（arr）+ 1， 第一个数和第二个数的差就应该为 d * len(arr)，

所以可以找到d，然后就可能得到等差数列中的所有数。

注意最后要处理一下差为0的edge case。

时间复杂度：O（n）

空间复杂度：O（1）

class Solution(object):
    def missingNumber(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        d = (arr[-1] - arr[0]) // len(arr)
        
        k = arr[0]
        for num in arr:
            if num != k:
                return k
            k += d
        return k