HRZ的序列

题目：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;
                }
            }
        }
    }
}