leetcode_1502 判断能否形成等差数列

要求

给你一个数字数组 arr 。

如果一个数列中，任意相邻两项的差总等于同一个常数，那么这个数列就称为 等差数列 。

如果可以重新排列数组形成等差数列，请返回 true ；否则，返回 false 。

示例 1：

输入：arr = [3,5,1]
输出：true
解释：对数组重新排序得到 [1,3,5] 或者 [5,3,1] ，任意相邻两项的差分别为 2 或 -2 ，可以形成等差数列。

示例 2：

输入：arr = [1,2,4]
输出：false
解释：无法通过重新排序得到等差数列。

提示：

• 2 <= arr.length <= 1000
• -10^6 <= arr[i] <= 10^6

核心代码

class Solution:
    def canMakeArithmeticProgression(self, arr: List[int]) -> bool:
        l = len(arr)
        if l <= 2:
            return True
        arr.sort()
        for i in range(1, l - 1):
            if arr[i] * 2 != arr[i - 1] + arr[i + 1]:
                return False
        return True

第二种解法

class Solution:
    def canMakeArithmeticProgression(self, arr: List[int]) -> bool:
        arr.sort()
        flag = arr[1]-arr[0]
        for i in range(len(arr)-1):
            print(arr[i],arr[i+1],flag)
            if arr[i+1] != (arr[i] + flag):
                return False
        return True

解题思路：此题比较简单，主要是前一项 + 后一项 = 本项 * 2,我们还可以使用排序后，循环遍历的方式，记录前后项差值，所有差值都相同就是等差数列。