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

124 阅读1分钟

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