CSP-M2 A-HRZ的序列

Limit

时间限制	1s
空间限制	64MB

题目描述

相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列aa，他对这个序列产生了浓厚的兴趣。

他好奇是否存在一个数K，使得一些数加上K，一些数减去K，一些数不变，使得整个序列中所有的数相等。

其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。

由于瑞神只会刷B站，所以他把这个问题交给了你！

输入格式

输入第一行是一个正整数tt表示数据组数。

接下来对于每组数据，输入的第一个正整数n表示序列aa的长度，随后一行有n个整数，表示序列a。

输出格式

输出共包含t行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。（输出不包含引号）

样例输入

2
5
1 2 3 4 5
5
1 2 3 4 5

样例输出

NO
NO

Note

数据点(上限)	t	n	ai
1,2	10	10	10
3,4,5	10	103	109
6,7,8,9,10	10	104	1015

思路

本题思路比较简单，将输入数据存入一个数组a中，然后对这个数组进行排序，排好序后对数组进行遍历，

1. 如果发现数组a中包含的不同数字个数超过三个直接输出NO并返回
2. 如果发现数组a中包含的不同数字个数低于三个直接输出YES并返回
3. 如果发现数组a中包含不同数字个数等于三个则判定最大，最小数与数值第二的数字的差值的绝对值：
   （1）如果差值的绝对值相等，则输出YES并返回
   （2）如果差值的绝对值不相等，则输出NO并返回

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        ll *a=new ll[n+10];
        for(int i=0;i<n;i++)
            scanf("%lld",&a[i]);
        sort(a,a+n);
        int co=1;
        ll b[4];
        b[0]=a[0];
        for(int i=0;i<n-1;i++)
        {
            if(a[i]!=a[i+1])
            {
                b[co]=a[i+1];
                co++;
                if(co>=4)
                    break;
            }
        }
        if(co>3)
            cout<<"NO"<<endl;
        else if(co==3&&(b[2]-b[1])!=(b[1]-b[0]))
        	cout<<"NO"<<endl;
        else
            cout<<"YES"<<endl;
    }
    return 0;
}