Wek8 CS-1

问题描述：

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

Input
输入第一行是一个正整数t表示数据组数。
接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。
Output
输出共包含t行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。（输出不包含引号）

Sample Input
2
5
1 2 3 4 5
5
1 2 3 4 5
1
2
3
4
5
Sample Output
NO
NO

解题思路：

首先将数组利用sort排序。以第一项a[0]分类。符合条件共有三种情况：（设ans为最后使整个数列相等的数）
①ans=a[0]不变或者ans=a[0]-k，则ans=a[0]，则所有的数都要等于a[0]。即序列所有项都相等。
②a[0]＋k，则往右遍历数组，遇到第一个大于a[0]的数a[i]，如果该序列符合条件，ans=a[i]，此时k=a[i]-a[0]，继续往右遍历数组，所有的数都必须满a[j]=ans，或者ans+k=a[j]，否则，该序列不符合条件。
note::!!错失50分，因为没有考虑数据范围，数组类型应该是long long int……

实验代码：

#include<iostream>
#include<algorithm>
using namespace std;
long long int a[100000];		
int size;		
void judge(int n)
{	 
	bool flag=1;			
	long long int k,ans=a[0];		
	int temp=n;			
	
	for(int i=1;i<n;i++)
	{
		if(a[i]>ans)
		{
			k=a[i]-ans;		
			temp=i;
			break;			
		}
	} 
	for(int i=temp+1;i<n;i++)	
	{
		if(a[i]-k!=ans)
		{
			flag=0;
			break;			
		}
	}
	if(flag)
	{
		cout<<"YES"<<endl;
		return ;			 
	}
	flag=1;
	temp=n,ans=a[0];		
	for(int i=1;i<n;i++)
	{
		if(a[i]>a[0])		
		{
			ans=a[i];
			k=ans-a[0];
			temp=i;	
			break;			 
		}
	}
	for(int i=temp+1;i<n;i++)
	{
		if(a[i]!=ans&&(a[i]-k)!=ans)
		{
			flag=0;
			break;
		}
	} 
	if(flag==1)
	{
		cout<<"YES"<<endl;
		return;
	}
	cout<<"NO"<<endl;
}
int main(void)
{
	int n,len;
	cin>>n;		
	for(int j=0;j<n;j++)
	{
		cin>>len;		
		for(int i=0;i<len;i++)
		{
			cin>>a[i];		
		}
		sort(a,a+len);		
		judge(len);
	} 
	return 0;
}