LeetCode|1502.判断能否形成等差数列

最新推荐文章于 2022-08-02 15:48:13 发布

xiao黄

最新推荐文章于 2022-08-02 15:48:13 发布

阅读量430

点赞数

分类专栏： LeetCode 文章标签： leetcode python

本文链接：https://blog.csdn.net/Python_Matlab/article/details/108004166

版权

LeetCode 专栏收录该内容

76 篇文章

订阅专栏

LeetCode习题答案汇总

题目：
给你一个数字数组 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:
        nums = set()
        arr.sort()
        for i in range(len(arr)-1):
            a = abs(arr[i+1]-arr[i])
            nums.add(a)
            if len(nums) > 1:
                return False
        return True