序列问题 (鞭尸 long long).c++

问题描述：
有一个序列a，问是否存在一个数K，序列中的一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等。
其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作

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

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

sample input:
1
5
1 2 3 4 5

sample output:
NO

数据点（上限）	t	n	ai
1,2	10	10	10
3,4,5	10	103	10 9
6,7,8,9,10	10	104	1015

思路：

• 由于最大的数据范围是10 ¹⁵,所以一定要使用 long long ,这道题的数据给的很随意，就导致我本来用的int ,想着无论怎么样都能过上几个点拿个基本分，结果直接翻车，成绩凄凉，所以一定要用long long
• 由于题目说明一些数+K,一些数-K，一些数不变然后相等，就说明这个序列不管多长，必须只有三种数
• 所以可以先将所有的元素读入之后排序，将其中不同的元素取出到一个新的数组中
• 如果序列中只有一种或者两种不同的数，那么直接输出YES
• 如果序列中有超过三种不同的数，直接输出NO
• 如果序列中有三种不同的数，那就比较相邻两个元素之间的差是否相等，相等输出YES

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int N=1e5+5;
int main()
{
 int t,n;
 long long a[N];
 scanf("%d",&t);
 while(t--)
 {
  scanf("%d",&n);
 for(int i=0;i<n;i++)
  scanf("%lld",&a[i]);
 sort(a,a+n);
 long long tot=1;
 long long ans[N];
 ans[0]=a[0];
 for(int i=1;i<n;i++)
 {
  if(a[i]!=a[i-1])
  { 
   ans[tot]=a[i];
   tot++;
  }
 }
 if(tot>3) printf("NO\n");
 if(tot<=2) printf("YES\n");
 else if(tot==3)
 {
  long long temp1=ans[1]-ans[0];
  long long temp2=ans[2]-ans[1];
  if(temp1==temp2) printf("YES\n");
  else printf("NO\n");
 }
 }
 return 0;
}