LeetCode|1502.判断能否形成等差数列

LeetCode习题答案汇总

题目：
给你一个数字数组 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:
        nums = set()
        arr.sort()
        for i in range(len(arr)-1):
            a = abs(arr[i+1]-arr[i])
            nums.add(a)
            if len(nums) > 1:
                return False
        return True

结果：