题意:
题目描述
相较于咕咕东,瑞神是个起早贪黑的好孩子,今天早上瑞神起得很早,刷B站时看到了一个序列 ,他对这个序列产生了浓厚的兴趣,他好奇是否存在一个k数 ,使得一些数加上k ,一些数减去 k,一些数不 变,使得整个序列中所有的数相等,其中对于序列中的每个位置上的数字,至多只能执行一次加运算或 减运算或是对该位置不进行任何操作。由于瑞神只会刷B站,所以他把这个问题交给了你!
输入格式
输入第一行是一个正整数 表示数据组数。 接下来对于每组数据,输入的第一个正整数 表示序列 的长 度,随后一行有 个整数,表示序列 。
输出格式
输出共包含 行,每组数据输出一行。对于每组数据,如果存在这样的K,输出"YES",否则输出“NO”。 (输出不包含引号)
样例输入
2
5
1 2 3 4 5
5
1 2 3 4 5
样例输出
NO
NO
思路:
若是经历过某些数加上K,某些数减去K ,某些数不变的操作后所有的数都相等,那么数列中最多只有三个不同的数。若是出现了三个以上的数则一定不存在。当值出现一个或两个数时,一定成功,而当三个数的时候则要满足大小位于中间的那个数是最小的数和最大的数的中位数,只有这样才能满足最小的数+K = 最大的数 - K = 大小位于中间的数。
代码:
#include<stdio.h>
#include<algorithm>
long long number[10005];
int main()
{
int t;
scanf("%d", &t);
while(t--!=0)
{
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%lld", &number[i]);
long long x[3];
int tot = 0;
bool onelayer = true;
for (int i = 0; i < n && onelayer; i++)
{
bool judge = true;
for (int k = 0; k < tot && judge; k++)
{
if (number[i] == x[k])
judge = false;
}
if (judge)//如果还没有出现过
{
if (tot > 2)onelayer = false;
else
{
x[tot] = number[i];
tot++;
}
}
}
if (onelayer)
{
if (tot == 2 || tot == 1 )printf("YES\n");
else //只有三个数
{
std::sort(x, &x[3]);
if(x[0] + x[2] == 2*x[1])printf("YES\n");
else printf("NO\n");
}
}
else
{
printf("NO\n");
}
}
}