题目:HRZ的序列
题意:相较于咕咕东,瑞神是个起早贪黑的好孩子,今天早上瑞神起得很早,刷B站时看到了一个序列 ,他对 这个序列产生了浓厚的兴趣,他好奇是否存在一个数 ,使得一些数加上 ,一些数减去 ,一些数不 变,使得整个序列中所有的数相等,其中对于序列中的每个位置上的数字,至多只能执行一次加运算或 减运算或是对该位置不进行任何操作。由于瑞神只会刷B站,所以他把这个问题交给了你!
输入:
输入第一行是一个正整数 表示数据组数。 接下来对于每组数据,输入的第一个正整数 表示序列 的长 度,随后一行有 个整数,表示序列 。
输出:
输出共包含 行,每组数据输出一行。对于每组数据,如果存在这样的K,输出"YES",否则输出“NO”。 (输出不包含引号)
样例:
解题思路:这个题看似很麻烦,其实就是找有几个不同点;如果小于3个,直接yes;等于3个,判断一下是否是等差队列;大于3个直接no。
代码:
#include<iostream>
#include<algorithm>
#define Max 1000000000000005
using namespace std;
int main()
{
long long p;
cin>>p;
while(p--)
{
long long n,x;
cin>>n;
long long total=0,t[4];
t[0]=t[1]=t[2]=t[3]=Max;//这里弄3个值
for(long long i=0;i<n;i++)
{
cin>>x;
int flag=0;
for(long long j=0;j<4;j++)
{
if(t[j]==x)//如果有重复的,打个标记
{
flag=1;
break;
}
}
if(!flag)
{
if(total<4)//没用重复的,+1
{
t[total]=x;
}
total++;
}
}
if(total>=4)//大于3个
{
cout<<"NO"<<endl;
}else
{
if(total<3)//小于3个
{
cout<<"YES"<<endl;
}else
{
sort(t,t+3);
if(t[1]*2==t[0]+t[2])//等于3个
{
cout<<"YES"<<endl;
}else
{
cout<<"NO"<<endl;
}
}
}
}
}