XTU online judge 1273 Set

Description
对于包含整数多值集合A={a(1),a(2),⋯,a(n) } ，执行以下两步操作
1.将某些元素加上整数x
2.将某些元素减掉整数y
请问能否使得A 的元素全部相等？

Input
第一行是一个整数K(1≤K≤100)，表示样例的个数。
每个样例的第一行是一个整数n(1≤n≤100,000) 。
第二行是{a(i) | i=1,2,⋯n},0≤ a(i) ≤1,000,000,000 。

Output
每行输出一个样例的结果，如果可以使得元素全部相等输出"Yes",否则输出"No"。

Sample Input
2
5
2 1 2 1 3
5
1 2 3 4 5

Sample Output
Yes
No

解题思路：
判断并存储集合A的元素种类数，大于4肯定不可能实现元素全部相等，小于等于3肯定可以，而等于4的情况需要特殊判断一下。

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int k;
	cin>>k;
	while(k--)
	{
		int n;
		int c[10];
		memset(c,0,sizeof(c));
		
		int m,j=0,sum=1;
		cin>>n;
		scanf("%d",&m);
		c[0]=m;
		//判断并存储元素种类数
		for(int i=1; i<n; i++)
		{
			scanf("%d",&m);	
			if(sum <= 4)
			{
				for(j=0; j<sum; j++)
				{
					if(m == c[j]) break;
				}
				if(j >= sum)
				{
					//c数组里存储的是去重后的集合元素
					c[sum]=m;
					sum++;
				}
			}
		}
			
		if(sum <= 3) cout<<"Yes"<<endl;
		else if(sum == 4)
		{
			if(c[0]+c[1] == c[2]+c[3] || c[0]+c[2] == c[1]+c[3] || c[0]+c[3] == c[1]+c[2])
				cout<<"Yes"<<endl;
			else
				cout<<"No"<<endl;
		}
		else
			cout<<"No"<<endl;
	}
	return 0;
}