csp模拟题

A - 题描述

相较于咕咕东，瑞神是个起早贪黑的好孩子，今天早上瑞神起得很早，刷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

解题思路

本题题目理解起来很简单，也没必要使用什么算法。直接输入三个数字，然后判断这三个数字相同的个数有几个，若三个均不相同，则依次计算两个差值是否相同，相同则输出YES，不同则输出NO。若三个中有相同的，则直接输出YES。但是却有几个小小的易错的坑，比如我刚开始理解错了题意，认为“2 2 3”这样的示例应该输出NO。再比如注意到数据要求，需要用long long 开数据，否则会不够用。只要注意到了这个问题，题目解决起来就很容易了。

代码

#include<iostream>
#include<algorithm>
using namespace std;
const int maxm=1*1e4+10;
int t,n;
long long sum,a[maxm],d[2],differ;
bool ans;
int main()
{
   
	std::ios::sync_with_stdio(false);//关掉同步  
    cin>>t;
    for(int i=0;i<t;i++)
    {
   
    	d[0]=d[1]=0;
    	differ=1;//初始化 
    	ans=true;
    	cin>>n;
    	for(int j=0;j<n;j++)
    		cin>>a[j];
   		sort(a,a+n);
   		sum=a[0];
   		int y=0;
   		for(int j=0;j<n;j++)
   		{
   
   			if(a[j]!=sum) 
   			{
   
   				d[y++]=a[j]-sum;//记录差值 
   				sum=a[j];
			    differ++;
		    }
	        if(differ>3) {
    ans=false; break; }
	        
	    }
	    if(ans && ((differ==3&&d[0]==d[1])||(differ<3))