山东大学程序设计思维CSP模拟2

HRZ的序列

题目描述 #
相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷B站时看到了一个序列 ，他对这个序列产生了浓厚的兴趣，他好奇是否存在一个数 ，使得一些数加上 ，一些数减去 ，一些数不变，使得整个序列中所有的数相等，其中对于序列中的每个位置上的数字，至多只能执行一次加运算或减运算或是对该位置不进行任何操作。由于瑞神只会刷B站，所以他把这个问题交给了你！

输入格式 #
输入第一行是一个正整数 表示数据组数。 接下来对于每组数据，输入的第一个正整数 表示序列 的长度，随后一行有 个整数，表示序列 。
输出格式 #
输出共包含 行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。
（输出不包含引号）
样例输入 # 样例输出 # 25
1 2 3 4 5
5
1 2 3 4 5
NO
NO

分析：签到题，稍微想一下就ok

//
// Created by haofeng on 4/8/20.
//
#include <iostream>
#include <algorithm>
#include <stdint.h>
using namespace std;
int T,N;
long long num[10010];
int main(){
   
    cin>>T;
    for(int i=0;i<T;i++){
   
        cin>>N;
        for(int j=1;j<=N;j++){
   
            scanf("%lld",&num[j]);
        }
        long long group[3];int fill=0;
        for(int j=0;j<3;j++)group[j]=0;
        bool K= true;
        for(int j=1;j<=N;j++){
   
            if(fill!=3){
   
                bool cp=false;
                for(int k=0;k<=fill;k++){
   
                    cp|=(group[k]==num[j]);
                }
                if(!cp){
   group[fill]=num[j];fill++;}
            }
            else{
   
                if(num[j]==group[0]||num[j]==group[1]||num[j]==group[2]){
   
                    continue;
                }
                else{
   
                    K=false;
                    break;
                }
            }
        }
        sort(group,group+3);
        if(K&&(