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
解释：无法通过重新排序得到等差数列。

代码

python

class Solution:
    def canMakeArithmeticProgression(self, arr: List[int]) -> bool:
        n = len(arr)
        arr.sort()			# 对arr排序

        a = arr[1] - arr[0]	# 记录等差的差值
        
        num = 0
        # 记录满足数列的差值个数
        for i in range(1, n):
            if (arr[i] - arr[i-1]) == a:
                num += 1
		# 判断是否满足等差数列差值个数，满足返回True，否则返回False
        if num == n-1:
            return True
        else:
            return False

执行结果：通过
执行用时：40 ms
内存消耗：13.6 MB