程序设计思维与实践 CSP-M2 A - HRZ 的序列

题目描述
相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列a，他对这个序列产生了浓厚的兴趣。
他好奇是否存在一个数KK，使得一些数加上KK，一些数减去K，一些数不变，使得整个序列中所有的数相等。
其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。
由于瑞神只会刷B站，所以他把这个问题交给了你！
输入格式
输入第一行是一个正整数tt表示数据组数。
接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。
输出格式
输出共包含tt行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。（输出不包含引号）
样例输入
2
5
1 2 3 4 5
5
1 2 3 4 5
样例输出
NO
NO
时间限制 1s
空间限制 64MB
数据点(上限) t n a_i
1,2 10 10 10
3,4,5 10^3 10^9
6,7,8,9,10 10 10^4 10^15
解题思路：
创建一个vector数组a，对于每个样例先输入n个数，当数组中不同数大于3时，就输出NO。当不同数的个数为1或2时，输出YES。当为3时，判断中间数是否为三个数的中位数。这里去重采用unique，注意需要先对数组排序。

sort(a.begin(),a.end(),cmp);
		  a.erase(unique(a.begin(), a.end()), a.end());

最后需要注意数组类型为long long类型。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
bool cmp(long long a,long long b)
{
	return a<b; 
}
int main()
{
	int t,n;
	long long m;
    vector<long long> a;
	scanf("%d",&t);
	while(t--)
	{
		a.clear();
		scanf("%d",&n);
		
		for(int i=0;i<n;i++)
		{
			scanf("%lld",&m);
			a.push_back(m);
		}
		sort(a.begin(),a.end(),cmp);
		  a.erase(unique(a.begin(), a.end()), a.end());
		if(a.size()>3)  printf("NO\n");
		else
		{
			if(a.size()==1||a.size()==2)  printf("YES\n");
			else
			 {
			 
			if((a[0]+a[2])==2*a[1])
			   printf("YES\n");
			else
			   printf("NO\n");
		     }
		}
	 } 
	return 0;
 }