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

题目链接:
https://leetcode-cn.com/problems/can-make-arithmetic-progression-from-sequence

解题思路:
1.对给的列表排序;
2.求出前两项的差值;
3.依次遍历,判断后面相邻两项的差值是否等于这个差值,等于就继续,不等于就返回False
4.如果遍历完成了,都没有返回False,那么就返回True。

代码:

class Solution:
    def canMakeArithmeticProgression(self, arr):
        new_arr = sorted(arr)
        minus_value = new_arr[1] - new_arr[0]
        for i in range(1,len(new_arr)):
            if new_arr[i] - new_arr[i-1] == minus_value:
                continue
            else:
                return False
        return True

s = Solution()
print(s.canMakeArithmeticProgression([3,5,1]))

猜你喜欢

转载自blog.csdn.net/annlin2009/article/details/114129910