【力扣 1502】判断能否形成等差数列 C++题解（数学+排序）

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

---

思路

首先，函数使用sort函数将arr排序。然后，计算数组中前两个元素的差diff，这个差值应该等于等差数列的公差。

然后，函数使用一个迭代器it1遍历arr的每一个元素（从第二个元素开始）。在每次循环中，如果当前元素和前一个元素的差不等于diff，那么就返回false，因为这个数组不能形成等差数列。

如果遍历完所有元素都没有返回false，那么就返回true，表示arr可以形成等差数列。

---

AC代码

/*
 * @lc app=leetcode.cn id=1502 lang=cpp
 *
 * [1502] 判断能否形成等差数列
 */

// @lc code=start
class Solution {
   public:
	bool canMakeArithmeticProgression(vector<int>& arr) {
		sort(arr.begin(), arr.end());
		int diff = arr[1] - arr[0];
		auto it1 = arr.begin() + 1;
		for (; it1 != arr.end(); it1++) {
			if (*it1 - *(it1 - 1) != diff) {
				return false;
			}
		}
		return true;
	}
};
// @lc code=end