判断一组数列能否形成等差数列

计算数列中相邻的差值,如果差值都相等,则为等差数列。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

你可能感兴趣的:(python初学者,算法,python)