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;
    }
}

你可能感兴趣的:(LeetCode题解之判断能否形成等差数列)