力扣 5452. 判断能否形成等差数列

给你一个数字数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-make-arithmetic-progression-from-sequence
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

先排序，然后计算第一项和第二项的差，遍历所有项，判断前后项的差是否相等，只有遇到一个不相等即可返回false，否则返回true。
对于只有数组长度为1或2的可以直接返回true。

class Solution {
public:
    bool canMakeArithmeticProgression(vector<int>& arr) {
        
        sort(arr.begin(),arr.end());
        int diff = 0;
        if(arr.size()>=2) 
            diff = arr[1] - arr[0];
        for(int i = 2;i < arr.size();i++)
        {
            if(arr[i] - arr[i-1] != diff)
                return false;
        }
        return true;
    }
};