计算数列中相邻的差值,如果差值都相等,则为等差数列。set函数可以去掉重复的元素,当集合的长度为1,则为等差数列。
示例 1:
输入:arr = [3,5,1]
输出:true
解释:对数组重新排序得到 [1,3,5] 或者 [5,3,1] ,任意相邻两项的差分别为 2 或 -2 ,可以形成等差数列。
示例 2:
输入:arr = [1,2,4]
输出:false
解释:无法通过重新排序得到等差数列。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/can-make-arithmetic-progression-from-sequence
class Solution(object):
def canMakeArithmeticProgression(self, arr):
"""
:type arr: List[int]
:rtype: bool
"""
arr.sort()
a=[]
j=1
for i in range(len(arr)):
c=arr[i]-arr[j]
j+=1
a.append(c)
if j>=len(arr):
break
if len(set(a))==1:
return True
else:
return False