CSP-M2

HRZ的序列

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

做法

首先考虑比较简单的情况：少于两个数时，一定符合要求。有三个数时，那么这三个数升序排列后，最小和最大的数相加一定是2的整数倍，并且是中间数的二倍。
接下来考虑数据比较多的情况，其实和前面简单数据差不多。可以排序后利用前缀和，数列要想符合要求，不同的数最多只能有3个，也就是前缀和不为0的不能多于3个；如果少于3个，就一定符合要求；恰好为3个，还是需要满足最小和最大的数相加一定是2的整数倍，并且是中间数的二倍。

代码

#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
long long a[100000];

int main()
{
   
	int t;
	cin>>t;
	for(int it=0;it<t;it++)
	{
   
		int n;
		scanf("%d",&n);
		for(int i=0;i<n;i++)
		{
   
			cin>>a[i];
		}
		sort(a,a+n);
		
		if(n<=2)
		cout<<"YES"<<endl;
		else if(n==3)
		{
   
			if((a[0]+a[2])/2==a[1]&&(a[0]+a[2])%2==0)
			{
   
				cout<<"YES"<<endl;
			}
			else
			cout<<"NO"<<endl;
		}
		else
		{
   
			long long c,mid;
			int num=1;
			for(int i=1;i<n;i++)
			{
   
				c=a[i]-a[i-1];
				if(c!=0)
				num++;
				if(num==2)
				mid=a[i];
			}
			if(num>3)
			{
   
				cout<<"NO"<<endl;
			}
			else if(num