week8 实验A HRZ的序列

题目：
相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列 ，他对 这个序列产生了浓厚的兴趣，他好奇是否存在一个数K ，使得一些数加上K ，一些数减去 K，一些数不变，使得整个序列中所有的数相等，其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。由于瑞神只会刷B站，所以他把这个问题交给了你！
输入：
输入第一行是一个正整数 表示数据组数。 接下来对于每组数据，输入的第一个正整数 表示序列 的长 度，随后一行有 个整数，表示序列 。
输出：
输出共包含 行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。 （输出不包含引号）
样例输入：
2
5
1 2 3 4 5
5
1 2 3 4 5
样例输出：
NO
NO

在课上的时候没有想到用set，而且判断条件漏写了一句，直接爆零了。
课后知道了用set一下就过了。
set可以自动去重，那么我们只需要将数据全部放入set，如果最后set的size小于3，那么肯定存在K使得两个数相等，输出“YES"；如果set的size大于3，则肯定不存在K；如果size等于3，因为set自动排序，则将数据放到a[0],a[1],a[2]中，看a[1]-a[0]是否等于a[2]-a[1]，如果相同，输出YES，否则输出NO。
以下是完整代码：

#include<iostream>
#include<set>
#include<algorithm>
using namespace std;
set<long long> s;
int t;
int n;
long long a[3];
int main()
{
 cin>>t;
 for(int cnt=0;cnt<t;cnt++)
 {
  cin>>n;
  long long temp;
  for(int i=0;i<n;i++)//输入 
  {
   cin>>temp;
   s.insert(temp);
  }
  if(s.size()<3)
  {
   cout<<"YES"<<endl;
  }
  else if(s.size()==3)
  {
   int j=0;
   for(set<long long>::iterator i=s.begin();i!=s.end();i++)//把数据拿出来 
   {
    a[j]=*i;
    j++;
   }
   if((a[1]-a[0])==(a[2]-a[1]))
   {
    cout<<"YES"<<endl;
   }
   else
   {
    cout<<"NO"<<endl;
   }
  } 
  else
  {
   cout<<"NO"<<endl;
  }
  s.clear();
 }
}