网络异常,图片无法展示
|
题目
给你一个数字数组 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)