CSP-M2补题_A HRZ的序列

题目描述

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

输入输出

Input

输入第一行是一个正整数 表示数据组数。 接下来对于每组数据，输入的第一个正整数 表示序列 的长 度，随后一行有 个整数，表示序列 。

Output

输出共包含 行，每组数据输出一行。对于每组数据，如果存在这样的K，输出"YES"，否则输出“NO”。 （输出不包含引号）

Sample Input

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

Sample Output

NO
NO

思路分析

这道题仔细想想可以将数据的分为几类：
序列中的数全部相同，那么加减0就满足，一定存在；序列中有两种不同的数，那么加减二者差的一半就满足条件，一定存在；序列中数的种类大于3，无论如何也不满足，不存在这样的数k；序列中的数有三种，则必须是等差数列，此处用set存储，利用互异性，减少数据量，只需要操作三个数。

总结反思

此题考试的时候想到了多组数据需要清空，但是忘记了set中最后一个数是.end()-1，而不是.end()。恰恰用.end()样例过了…大意了…

代码

#include<iostream>
#include<stdio.h>
#include<set>
using namespace std;

int t,n;
const int maxn=1e4+5;
long long a[maxn];
set<long long> s;//利用集合元素互异性
 
int main()
{
    cin>>t;
    for(int i=0;i<t;i++)
    {
        cin>>n;
        for(int j=0;j<n;j++)
            a[j]=0;
        s.clear();
        for(int j=0;j<n;j++)
        {
            cin>>a[j];
            s.insert(a[j]);
        }
        if(s.size()>3) 
            cout<<"NO"<<endl;
   	else if(s.size()==3)
 	{
   		set<long long>::iterator it=s.begin();
   		long long tmp=*it;
   		it++;
   		it++; 
   		tmp+=*it;
   		it--;
   		long long mid=*it;
   		if(2*mid!=tmp) cout<<"NO"<<endl;
   		else cout<<"YES"<<endl; 
  	}
    	else if(s.size()<3)
   		cout<<"YES"<<endl;  
    }
    return 0;
}