5452. 判断能否形成等差数列（Leetcode 196周赛）

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

思路很简单，对向量进行排序后，计算前两个的差值，并比较接下来的差值是否域值相等即可。

// 这个cmp定义需要为全局
int cmp(const void *a, const void *b)
{
	return *(int*)a > *(int*)b;
}
class Solution {
public:
	bool canMakeArithmeticProgression(vector& arr) {
		qsort(&arr[0], arr.size(), sizeof(int), cmp);
		int K = arr[1] - arr[0];
		for (int i = 2; i < arr.size(); ++i) {
			if (arr[i] - arr[i - 1] != K)
				return 0;
		}
		return 1;
	}
};

另一种sort函数使用：

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

上面时利用C++自带的排序算法实现的，下面用自己的快排实现：

class Solution {
	void myqsort(vector& ps, int l, int r)
	{
		if (l < r)
		{
			int i = l, j = r;
			int tmp = ps[i];
			int key = ps[i];

			while (i < j)
			{
				while (i < j && key <= ps[j]) --j;
				ps[i] = ps[j];
				while (i < j && ps[i] < key) ++i;
				ps[j] = ps[i];
			}

			ps[i] = tmp;

			myqsort(ps, l, i - 1);
			myqsort(ps, i + 1, r);
		}
	}
public:
	bool canMakeArithmeticProgression(vector& arr) {
		myqsort(arr, 0, arr.size() - 1);
		int K = arr[1] - arr[0];
		for (int i = 2; i < arr.size(); ++i) {
			if (arr[i] - arr[i - 1] != K)
				return 0;
		}
		return 1;
	}
};