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

解题思路

先对数组进行排序，然后迭代去判断每两个相邻的元素的差值是否相同。

复杂度分析

• 时间复杂度：O(nlogn)，排序的时间复杂度为O(nlogn)。
• 空间复杂度：O(logn)，快排的空间复杂度为O(logn)。

代码实现

class Solution {
    public boolean canMakeArithmeticProgression(int[] arr) {
        Arrays.sort(arr);
        boolean res = true;
        for (int i = 0; i < arr.length - 2; i ++) {
            if (arr[i+2] - arr[i+1] != arr[i+1] - arr[i]) {
                res = false;
                break;
            }
        }
        return res;
    }
}