HRZ的序列

问题描述：

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

input：
输入第一行是一个正整数t表示数据组数。 接下来对于每组数据，输入的第一个正整数n表示序列a的长度，随后一行有n个整数，表示序列a。
output：
输出共包含t行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。 （输出不包含引号）

样例输入：

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

样例输出：

NO
NO

解题思路：

对于这个题，如果存在一个数K，使得一个数列中的数有的加上K，有的减去K，有的数不变，最终这些所有数相等。那么输入的数列中去重之后的数的个数必定是小于等于3个，如果等于3，这三个数必须构成一个等差数列。由此，我们可以通过stl中的set，因为set有去重和自动排序功能。在所有的数输入完之后，只需要判断set对象中元素的个数，如果大于3，则输出NO，如果小于3，则输出YES，如果等于3，还需要进一步判断这三个数是不是等差序列，如果是的话，则输出YES，否则输出NO。

代码：

#include<iostream>
#include<set>
#include<vector>
using namespace std;
int main()
{
	int t;
	cin>>t;
	for(int i=0;i<t;i++)//输入的数据组数
	{	
		int n,K;
		cin>>n;
		set<long long int> num;
		for(int i=0;i<n;i++)
		{
			long long int temp;
			cin>>temp;//将所有数据存在num中，自动去重
			num.insert(temp);
		}
		if(num.size()>3)//如果不重复的数字个数大于3，不存在K
			cout<<"NO"<<endl;
		else if(num.size()==3)//若等于3，则判断是不是等差数列
		{
			int count=0;
			int all_num=num.size();
			long long int *p=new long long int [all_num+1];
			set<long long int>::iterator it;
			for(it=num.begin();it!=num.end();it++)
			{
				count++;
				p[count]=*it;
			}
			if(p[3]-p[2]==p[2]-p[1])
				cout<<"YES"<<endl;
			else
				cout<<"NO"<<endl;
			delete [] p;	
		}
		else if(num.size()<3)//若小于3，则输出YES
			cout<<"YES"<<endl;
			
	}
	return 0;
 }