CSP-M2-补题-A-HRZ的序列

**

题目：

个人思路：

• 首先，在读完题目以后。先思考的问题是什么样的序列存在k，什么样的不存在k。

• 想了一会后，我发现，当序列中相异的数字个数cnt大于3时，那么肯定不存在k(可以证明)

• 所以当cnt大于三时，直接输出NO；

• 当cnt小于等于三的情况：

   1、cnt == 3，如果存在k满足条件，那么必定有：最小的数+最大的数 = 2*中间的数；如果不满足条件，说明不存在k；
   2、cnt == 2时，那么必定存在k满足条件；
   3、cnt == 1时，满足条件。
  

• 具体细节：

   1、sign[1]-sign[3]存储相异数字;
   2、cnt == 3 sort(sign+1,sign+4);在进行判断是否满足条件
   3、cnt == 1 || cnt == 2;满足条件;
  

---

代码块：

#include<iostream>
#include<algorithm>
using namespace std;
int t, n;
long long a[10010];
long long sign[4] = { -1 };
int main() {
	cin >> t;
	for (int s = 0; s < t; ++s) {
		cin >> n;
		int cnt = 0;
		for (int i = 0; i < 4; ++i)sign[i] = -1;//init
		for (int i = 1; i <= n; ++i) {
			cin >> a[i];
			bool flag = false;
			if (cnt <= 3)//对sign的处理
			{
				for (int j = 1; j < i; ++j) {
					if (a[i] == a[j]) {
						cnt--; flag = true;
						break;
					}
				}
				cnt++;
				if(!flag)sign[cnt] = a[i];
			}
		}
		sort(sign + 1, sign + 4);
		if (cnt > 3)cout << "NO" << endl;
		else if (cnt == 1 || cnt == 2)cout << "YES" << endl;
		else if (cnt == 3 && 2 * sign[2] == sign[1] + sign[3])cout << "YES" <<endl;
		else cout << "NO" << endl;
	}
	return 0;
}

---