第三次模拟测试-A

题目描述：
相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列a，他对这个序列产生了浓厚的兴趣。
他好奇是否存在一个数K，使得一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等。
其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。
由于瑞神只会刷B站，所以他把这个问题交给了你！
输入：
输入第一行是一个正整数t表示数据组数。
接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。
输出：
输出共包含t行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。（输出不包含引号）
sample：
input：
2
5
1 2 3 4 5
5
1 2 3 4 5
output：
NO
NO
题目分析：
此题表面上复杂，实际上它限制住了只允许有一个数进行加减，所以难度瞬间下降，为了方便，我们就可以把一组数排列之后再看他们的情况，毕竟它不要求是谁相加，而是只看结果

		sort(a,a+n);

分为几种情况：
如果都是一样的数，那一定可以，不动即可。
如果有只有两种不一样的数，那也可以，要么是一个加，要么是一个减。

		if(count==1||count==0)
		{
			cout<<"YES"<<endl;
		}

如果有三种不一样的数，那么就要记录下来看看两者的差是不是一样，如果一样就可以实现（一个加一个减），否则就不行

		if(count==2)
		{
			if(b[1]-b[0]==b[2]-b[1])
			{
				cout<<"YES"<<endl;
			}
			else
			{
				cout<<"NO"<<endl;
			}
		}

如果超过三个数不一样那就一定不可能了

		for(int i=0;i<n;i++)
		{
			if(b[count]==a[i])
			{
				continue;
			}
			else
			{
				count++;
				b[count]=a[i];
				if(count==3)
				{
					cout<<"NO"<<endl;
					break;
				}
			}
		}

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
long long a[10010];
int main()
{
	int t;
	cin>>t;
	for(int ii=1;ii<=t;ii++)
	{
		memset(a,0,sizeof(a));
		int n;
		cin>>n;
		for(int i=0;i<n;i++)
		{
			cin>>a[i];
		}
		sort(a,a+n);
		long long b[5];
		b[0]=a[0];
		int count=0;
		for(int i=0;i<n;i++)
		{
			if(b[count]==a[i])
			{
				continue;
			}
			else
			{
				count++;
				b[count]=a[i];
				if(count==3)
				{
					cout<<"NO"<<endl;
					break;
				}
			}
		}
		if(count==1||count==0)
		{
			cout<<"YES"<<endl;
		}
		if(count==2)
		{
			if(b[1]-b[0]==b[2]-b[1])
			{
				cout<<"YES"<<endl;
			}
			else
			{
				cout<<"NO"<<endl;
			}
		}
	}
}