week8 csp模拟A HRZ的序列

HRZ的序列

给定一个n(n<=104)个元素的序列ai(|ai|<=1015 开long long) ，问能否一些数加上某个相同的值k，一些数减k,同时一些数不变，使得序列内所有元素的值相等。可以则输出"YES",不可以则输出"NO"

Input

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

Output

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

Sample Input

2
5
1 2 3 4 5
5
1 2 3 4 5

Sample Output

NO
NO

我的思路：
这道题是要求将序列中的数加、减一个数k或不变，是否可以使得这个序列的所有数字相同。其实，根据题目的要求，序列中最多有三个不同的数，即a-k,a,a+k。于是，我们可以利用这点来进行判断，当遇到第三个不同的数时，判断是否满足题目的+/-k的条件，满足则继续。当遇到第四个不同的数时，则明显不满足条件。

我的总结：
这道题整体来说还是比较容易的，只要找到了规律，就可以很快地解决。

我的代码：

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;

long long t,n,nod,a[3],b;
int main()
{
	scanf("%lld",&t);
	for(long long i=0;i<t;i++)
	{
		bool can=true;
		nod=0;
		scanf("%lld",&n);
		for(long long k=0;k<n;k++)
		{
			scanf("%lld",&b);
			if(can && nod == 3) can=false;
			if(can){
			if(k==0) 
			{
				a[nod]=b;
				nod++;
			}
			else
			{
				bool is=true;
				for(long long q=0;q<nod;q++)
				{
					if(b==a[q])
					{
						is=false;
						break;
					}
				}
				if(is) 
				{
					if(nod==2)
					{
						if(a[0]+a[1]!=2*b+3*(abs(a[0]-a[1])) && a[0]+a[1]!=2*b && a[0]+a[1]!=2*b-3*(abs(a[0]-a[1]))) can=false;
					}
					else{
					a[nod++]=b;
					}
				}
			}
		}
		}
		if(can) printf("YES");
		else printf("NO");
		if(i!=t-1) printf("\n");
	}
	return 0;
}