Week 8 CSP M2 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，只需要判断输入的这个序列中有几种数，对于输入的序列a进行去重操作，使用unique（）函数，对去重后的序列进行排序，对于新序列中的每个数进行判断，如果与中间的数差值均相等或者与中间的值相等，那么存在K，否则不存在K
（现在想想好像在去重之后直接判断有几种数，如果>3种，那么不存在K，如果是3种，那么判断一下差值是否相等，如果<3种，那么K肯定存在）

总结

首先并且是犯了多次的错误！！数据的范围！！应该用long long（就因为这个很悲催的wa了）
其次在第一边写的时候，对于序列没有去重，直接把原序列的中间位置的数当作是中位数了，这样是不对的😂

代码

#include <iostream>
#include <algorithm>
using namespace std;

int main()
{
	int t;
	cin>>t;
	for(int i=0;i<t;i++)
	{
		int n;
		cin>>n;
		long long a[n];
		for(int j=0;j<n;j++)
		{
			cin>>a[j];
		}
		
		sort(a,a+n);
		long long temp=unique(a,a+n)-a; 

		long long sub=a[temp/2]-a[0];
		int flag=1;
		for(int k=1;k<temp/2;k++)
		{
			if(a[temp/2]-a[k]==sub||a[temp/2]==a[k])
				continue;
			else flag=0;
		}
		for(int k=(temp+2)/2;k<temp;k++)
		{
			if(a[k]-a[temp/2]==sub||a[k]==a[temp/2])
				continue;
			else flag=0;
		}
		if(flag==0)
			cout<<"NO"<<endl;
		else 
			cout<<"YES"<<endl;	
	}	
 }