[CodeForces-714B] Filya and Homework 思维

题意：
给你一个含有 n 个数的数组, 问你是否存在一个 x, 使得这个数组中的某些数加上 x, 某些数减去 x 后所有数都相等.

注意：这里的x 不是从数组中选出的x 是任意一个x

思路：

如果序列中不同的数大于3个，则不可能
如果序列中不同的数为3个， 分别为a，b，c（a<b<c） 那么把与a相同的数都加上（b-a）变成b，把与c相同的数都减去（c-b）也变成b 那么序列中所有的数都变成了b 满足题意
如果序列中不同的数为2个，a，b 则 把与a相同的数都加上b-a或者把与b相同的数都加上a-b
如果序列中不同的数为1个 无需操作

`

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int mod=1e9+7;
const int maxn=1e5+5;
int main()
{
	int n,a[maxn],b[5];
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
	sort(a,a+n);
	int cnt=1,flag=1;
	b[0]=a[0];
	for(int i=1;i<n;i++)
	{
		if(a[i]!=a[i-1])
		{
			b[cnt++]=a[i];
		}
		if(cnt==4)
		{
			flag=0;
			break;
		}
	}
	if(!flag)
		cout<<"NO";
	if(cnt<=2)
		cout<<"YES";
	else if(cnt==3)
	{
		if(b[1]-b[0]==b[2]-b[1])
			cout<<"YES";
		else cout<<"NO";
	}
	return 0;
 }