每日leetcode

1502. 判断能否形成等差数列 - 力扣（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

思路

1. 这题考察的应该是排序函数的使用方法，如果没有很高的时间复杂度要求的话调包就行，如果要求比如说O(n)的时空间复杂度，那就可能需要桶排序了（遍历一遍数组通过最大值到最小值的差值设计桶，然后再遍历将对应的数字放到桶里，最后按桶返回）。
2. 排完序后先将前两个数的差值作为公差，进行遍历，如果某相邻两数的差不等于公差直接返回false，否则遍历完成返回true。

代码实现

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

复杂度分析

• 时间复杂度：排序的开销O(nlogn)。
• 空间复杂度：排序的栈空间O(logn)。