题目描述
相较于咕咕东,瑞神是个起早贪黑的好孩子,今天早上瑞神起得很早,刷B站时看到了一个序列a,他对这个序列产生了浓厚的兴趣。
他好奇是否存在一个数KK,使得一些数加上KK,一些数减去K,一些数不变,使得整个序列中所有的数相等。
其中对于序列中的每个位置上的数字,至多只能执行一次加运算或减运算或是对该位置不进行任何操作。
由于瑞神只会刷B站,所以他把这个问题交给了你!
输入格式
输入第一行是一个正整数tt表示数据组数。
接下来对于每组数据,输入的第一个正整数n表示序列a的长度,随后一行有n个整数,表示序列a。
输出格式
输出共包含tt行,每组数据输出一行。对于每组数据,如果存在这样的K,输出"YES",否则输出“NO”。(输出不包含引号)
样例输入
2
5
1 2 3 4 5
5
1 2 3 4 5
样例输出
NO
NO
时间限制 1s
空间限制 64MB
数据点(上限) t n a_i
1,2 10 10 10
3,4,5 10^3 10^9
6,7,8,9,10 10 10^4 10^15
解题思路:
创建一个vector数组a,对于每个样例先输入n个数,当数组中不同数大于3时,就输出NO。当不同数的个数为1或2时,输出YES。当为3时,判断中间数是否为三个数的中位数。这里去重采用unique,注意需要先对数组排序。
sort(a.begin(),a.end(),cmp);
a.erase(unique(a.begin(), a.end()), a.end());
最后需要注意数组类型为long long类型。
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
bool cmp(long long a,long long b)
{
return a<b;
}
int main()
{
int t,n;
long long m;
vector<long long> a;
scanf("%d",&t);
while(t--)
{
a.clear();
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%lld",&m);
a.push_back(m);
}
sort(a.begin(),a.end(),cmp);
a.erase(unique(a.begin(), a.end()), a.end());
if(a.size()>3) printf("NO\n");
else
{
if(a.size()==1||a.size()==2) printf("YES\n");
else
{
if((a[0]+a[2])==2*a[1])
printf("YES\n");
else
printf("NO\n");
}
}
}
return 0;
}