程序设计思维与实践 Week8 模拟赛 A HRZ的序列

题目描述：

相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列 ，他对 这个序列产生了浓厚的兴趣，他好奇是否存在一个数K，使得一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等，其中对于序列中的每个位置上的数字，至多只能执行一次加运算或 减运算或是对该位置不进行任何操作。由于瑞神只会刷B站，所以他把这个问题交给了你！

输入格式:

输入第一行是一个正整数 表示数据组数。 接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列 。

输出格式：

输出共包含 行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。 （输出不包含引号）

思路：

观察序列的特性，如果一个序列满足以下任意性质，则一定 存在 K：

1. 序列中不同的数只有三个，并且其中两个的和等于另一个数的两倍。
2. 序列中不同的数小于三个。

教训：由于两个条件下存在K，我在模拟赛中分别输出yes时，第二种情况的yes没有换行，90分没了！！！注意！！！

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
long long n,a[10100],b[10100];
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n;
		for(int i=1;i<=n;i++)
		cin>>a[i];
		sort(a+1,a+n+1);
		int tmp=0;
		memset(b,0,sizeof(b));
		b[++tmp]=a[1];
		for(int i=2;i<=n;i++)
		if(a[i]!=a[i-1])
		b[++tmp]=a[i];
		if(tmp==1||tmp==2)
		cout<<"YES"<<endl;
		else if(tmp==3&&b[1]+b[3]==(b[2]*2))
		cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	}
	return 0;
}