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

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

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

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

Sample Input：
2
5
1 2 3 4 5
5
1 2 3 4 5

Sample Output：
NO
NO

Hint：

思路：
本题基本的思想是这一组数如果有三个以上不同的数是无法实现的，因此只需要考虑数组中不同的数小于等于3的情况，1种或2种肯定可以，如果是3种，那么先进行排序，然后依次做减法，如果相等则可以实现，否则不行。

总结：
本题的数据量达到了1e15，因此需要用到long long类型，否则最后一个点无法通过。

代码：

#include<bits/stdc++.h>
using namespace std;
const long long maxn=1e5+5;
int t,n;
long long sum[10],a[maxn];
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=0;i<n;i++)
			scanf("%lld",&a[i]);
		sort(a,a+n);
		sum[0]=a[0];
		int count=1;
		bool flag=true;
		for(int i=1;i<n;i++){
			if(a[i]!=a[i-1])
				sum[count++]=a[i];
			if(count==4){
				flag=false;
				break;
			}
		}
		if(!flag)
			printf("NO\n");
		if(count<=2)
			printf("YES\n");
		else if(count==3){
			if(sum[1]-sum[0]==sum[2]-sum[1])
				printf("YES\n");
			else
				printf("NO\n");
		}
	}
	return 0;
}