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:
        # 先进行排序，然后通过错位zip压缩成一个等差元组（假如成立的情况下）
        arr.sort()
        zipList = list(zip(arr,arr[1:]))
        # print(zipList)
        resDict = dict()#用字典方式存取比较快速
        for x,y in zipList:
            resDict[x-y] = 1
        # print(resDict)
        return len(resDict) == 1#如果为等差，字典只有一个值



if __name__ == '__main__':
    arr = [3,5,1]
    ret = Solution().canMakeArithmeticProgression(arr)
    print(ret)