A-HRZ的序列

A-HRZ的序列

一、题目描述

二、思路与算法

首先分析判定依据，怎样就可以实现加减相同的数，使得所有数最后都相等？
因为需要令所有数字满足经相同的数字处理后值相等，那么这个序列最多出现三个数字：x、x+k、x-k。少的话，有1个数、2个数都明显可以满足。
一开始写的代码分类比较复杂，考虑各种情况，反而容易错。
简化的关键是unique函数的应用。
首先用sort函数对这个序列进行降序排序，再调用unique函数，返回值是指向第一个重复的数字的迭代器。
当不同的数字个数少于3时，输出YES。
当不同数字个数大于3时，输出NO。
当不同数字个数等于3时，判定第二个数是否为这三个不同数的平均数，若是，输出YES，否则输出NO。
至此，算法结束，只需要根据数据组数进行输入输出次数的控制。

三、代码实现

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

int t=0;//数据组数 
int n=0;//序列长度 
vector<long long> a(0);

int main(){
	scanf("%d",&t);
	for(int i=0;i<t;i++){
		scanf("%d",&n);
		for(int j=0;j<n;j++){
			long long temp;
			scanf("%lld",&temp);
			a.push_back(temp);
		}
		sort(a.begin(),a.end());
		int number=unique(a.begin(),a.end())-a.begin();
		if(number>3){printf("NO\n");	}
		else{
			if(number==3){
				if(a[0]+a[2]!=2*a[1]){	printf("NO\n");	}
				else{printf("YES\n");	}
			}
			else{printf("YES\n");		}
		}
		a.clear();
	}
	return 0;
}

四、经验与总结

1. 最开始的写法是，用一个vector数组存储整个序列中所有不同的数字，之后对vector的size（）进行分类和判定。
   sort（）函数搭配unique（）函数很容易替代这个繁琐的操作。
   所以应该多了解STL函数，这样可以很大程度上简化代码，代码简化后，出错可能性也降低。
2. 关于unique（）函数的知识：
   unique函数属于STL中比较常用函数，它的功能是元素去重。即”删除”序列中所有相邻的重复元素(只保留一个)。
   此处的删除，并不是真的删除，而是指重复元素的位置被不重复的元素给占领了。由于它”删除”的是相邻的重复元素，所以在使用unique函数之前，一般都会将目标序列进行排序。
   unique函数的函数原型如下：（参数类型都是迭代器）
   iterator unique(iterator it_1,iterator it_2);
   这种类型的unique函数是我们最常用的形式。其中这两个参数表示对容器中[it_1，it_2)范围的元素进行去重(注：区间是前闭后开，即不包含it_2所指的元素),返回值是一个迭代器，它指向的是去重后容器中不重复序列的最后一个元素的下一个元素。
   （参考这里C++STL中的unique函数解析）