WEEK8 csp模拟 A - HRZ的序列

A - HRZ的序列

题目描述

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

Input

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

Output

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

Sample Input

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

Sample Output

NO
NO

Note

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

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

题解

用v数组来记录a数组中有多少种数，循环遍历a数组中所有的数，如果a[i]不在v数组中且v中元素少于三个，则将a[i]对应的数放到v数组中；如果a[i]不在v数组中且v中元素等于三，则说明这个数组中出现了多于三种元素，则直接退出循环，并修改suc为0（suc是用来记录a数组是否少于三种元素的,初始化为1）；若a[i]在数组中则不进行任何操作。循环结束检查suc等于0(说明a数组中有多于3种元素)则直接输出"NO"，不等于0则对v数组排序，并检查是否符合递增规则，符合则输出"YES"，不符合则输出"NO"。
易错点：需要考虑到a数组中只有一种数或两种数的情况，在记录完数组v后检查v的长度是否小于3即可，小于三则直接输出"YES"并continue进行下一组。
注意数据范围！注意数据范围！注意数据范围！

代码

//数据开long long
//判断只有两个数的情况 
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
int t;
int n;
long long a[10100];
vector<long long>v;
int main(int argc, char** argv) {
	scanf("%d",&t);
	while(t--){
		v.clear();
		bool suc=1;
		scanf("%d",&n);
		for(int i=0;i<n;i++){
			scanf("%lld",&a[i]);
		}
		for(int i=0;i<n;i++){
			bool have=0;
			for(int j=0;j<v.size();j++){
				if(v[j]==a[i]){have=1;break;}
			}
			if(v.size()==3&&!have){suc=0;break;}
			else if(v.size()<3&&!have){
				v.push_back(a[i]);
			} 
		}
		if(v.size()<3){printf("YES\n");continue;}
		if(suc==0){printf("NO\n");}
		else{
			sort(v.begin(),v.end());
			if(v[1]-v[0]==v[2]-v[1]){
				printf("YES\n");
			}else{
				printf("NO\n");
			}
		}
	}
	return 0;
}