思路:
可以通过对题目的观察可以发现当一个输入的序列中,如果不相同的数字的个数大于3,则说明肯定无法完成题目所要求的操作,输出NO。如果不相同的数字个数等于3,则在这种情况下,只有较大的数减去一个数等于中间数,而较小数加上同一个数等于中间数时才能符合题目的要求,否则其他情况下无法符合。如果不相同的数小于3,则一定能够符合题目要求,输出YES。因此对输入的数进行排序,判断输入的数大小是否相等,出现不相等的数则flag++,并把不相等的数存入到一个数组中,之后进行上述的判断,得出结果。
代码:
#include<iostream>
//#include<vector>
#include<algorithm>
using namespace std;
long long a[10050];
long long b[5];
int main()
{
int t;
cin>>t;
//vector<long long> v;
while(t--)
{
//v.clear();
int flag=1;
int n;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a[i];
sort(a,a+n);
b[0]=a[0];
int num=0;
int io=0;
while(num<n)
{
if(a[num]!=a[++io])
{
num=io;
b[flag++]=a[num];
}
}
flag--;
if(flag<3)
{
cout<<"YES"<<endl;
}
else if(flag>3)
{
cout<<"NO"<<endl;
}
else
{
if(b[0]+b[2]==2*b[1])
{
cout<<"YES"<<endl;
}
else
{
cout<<"NO"<<endl;
}
}
}
return 0;
}