A - 题描述
相较于咕咕东,瑞神是个起早贪黑的好孩子,今天早上瑞神起得很早,刷B站时看到了一个序列 ,他对这个序列产生了浓厚的兴趣,他好奇是否存在一个数 ,使得一些数加上 ,一些数减去 ,一些数不变,使得整个序列中所有的数相等,其中对于序列中的每个位置上的数字,至多只能执行一次加运算或减运算或是对该位置不进行任何操作。由于瑞神只会刷B站,所以他把这个问题交给了你!
Input
输入第一行是一个正整数 表示数据组数。 接下来对于每组数据,输入的第一个正整数 表示序列的长度,随后一行有 个整数,表示序列 。
Output
输出共包含 行,每组数据输出一行。对于每组数据,如果存在这样的K,输出"YES",否则出“NO”。(输出不包含引号)
Sample Input
2
5
1 2 3 4 5
5
1 2 3 4 5
Sample Output
NO
NO
解题思路
本题题目理解起来很简单,也没必要使用什么算法。直接输入三个数字,然后判断这三个数字相同的个数有几个,若三个均不相同,则依次计算两个差值是否相同,相同则输出YES,不同则输出NO。若三个中有相同的,则直接输出YES。但是却有几个小小的易错的坑,比如我刚开始理解错了题意,认为“2 2 3”这样的示例应该输出NO。再比如注意到数据要求,需要用long long 开数据,否则会不够用。只要注意到了这个问题,题目解决起来就很容易了。
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int maxm=1*1e4+10;
int t,n;
long long sum,a[maxm],d[2],differ;
bool ans;
int main()
{
std::ios::sync_with_stdio(false);//关掉同步
cin>>t;
for(int i=0;i<t;i++)
{
d[0]=d[1]=0;
differ=1;//初始化
ans=true;
cin>>n;
for(int j=0;j<n;j++)
cin>>a[j];
sort(a,a+n);
sum=a[0];
int y=0;
for(int j=0;j<n;j++)
{
if(a[j]!=sum)
{
d[y++]=a[j]-sum;//记录差值
sum=a[j];
differ++;
}
if(differ>3) {
ans=false; break; }
}
if(ans && ((differ==3&&d[0]==d[1])||(differ<3))