SDU_week8_T1 HRZ的序列（set操作）

题目描述

题目分析

题目问给定一个整数序列，是否可以对任意一个数+K/-K/不变三种操作，使得所有数相同。翻译一下：最多有三种数字，当有三种数字时两两差值相同。
解决方法是先sort使其后面的数总比前面的数大，然后使用set记录差值，等于0就continue，不等于0就insert，最后分析set中的元素即可。

HITS

记得每次清空set，上次大模拟就是类似错误（每次重新赋值后没sort一类错误）

代码

#define _CRT_SECURE_NO_WARNINGS
#include<algorithm>
#include <iostream>
#include<cmath>
#include<cstdio>
#include<set>
using namespace std;

int t, n;
long long a[10009];
//long long b[10009];
set<long long>s;

int main()
{
    scanf("%d", &t);
	while (t--)
	{
		s.clear();//记得清空set，血的教训
		scanf("%d", &n);
		for (int i = 1; i <=n; i++)
		{
			scanf("%lld", &a[i]);
		}
		sort(a+1, a + n+1);
		if (n == 1)
		{
			printf("YES\n");
			continue;
		}
		for (int i = 2; i <= n; i++)
		{
			if (a[i] - a[1] == 0)continue;
			else s.insert(a[i] - a[1]);
		}
		if (s.size() < 2) printf("YES\n");
		else if (s.size() == 2)
		{
			if((*s.rbegin()) == 2* (*s.begin()))//s.end()指向最后一个元素的下一个，应该使用*s.rbegin()
				printf("YES\n");
			else printf("NO\n");
		}
		else printf("NO\n");
	}
	return 0;
}