Bear and Three Balls CodeForces - 653A

Bear and Three Balls CodeForces

题目链接:https://cn.vjudge.net/problem/CodeForces-653A

Limak is a little polar bear. He has n balls, the i-th ball has size ti.

Limak wants to give one ball to each of his three friends. Giving gifts isn’t easy — there are two rules Limak must obey to make friends happy:

No two friends can get balls of the same size.
No two friends can get balls of sizes that differ by more than 2.
For example, Limak can choose balls with sizes 4, 5 and 3, or balls with sizes 90, 91 and 92. But he can’t choose balls with sizes 5, 5 and 6 (two friends would get balls of the same size), and he can’t choose balls with sizes 30, 31 and 33 (because sizes 30 and 33 differ by more than 2).

Your task is to check whether Limak can choose three balls that satisfy conditions above.

Input
The first line of the input contains one integer n (3 ≤ n ≤ 50) — the number of balls Limak has.

The second line contains n integers t1, t2, …, tn (1 ≤ ti ≤ 1000) where ti denotes the size of the i-th ball.

Output
Print “YES” (without quotes) if Limak can choose three balls of distinct sizes, such that any two of them differ by no more than 2. Otherwise, print “NO” (without quotes).

Examples
Input
4
18 55 16 17
Output
YES
Input
6
40 41 43 44 44 44
Output
NO
Input
8
5 972 3 4 1 4 970 971
Output
YES
Note
In the first sample, there are 4 balls and Limak is able to choose three of them to satisfy the rules. He must must choose balls with sizes 18, 16 and 17.

In the second sample, there is no way to give gifts to three friends without breaking the rules.

In the third sample, there is even more than one way to choose balls:

Choose balls with sizes 3, 4 and 5.
Choose balls with sizes 972, 970, 971.

题意:n个球,第i个球的大小为t[i].从中选出3个球来,要求任意两个球的大小不能一样,且任意两个球的大小之差不能超过2.(可以等于2),问有没有选择的方案。

分析:排序删除数组中的重复元素

code:

#include 
#include 
#include 
using namespace std;
int main()
{
     
	int flag = 0;
	vector<int> v;
	int number;
	cin >> number;
	for(int i = 0; i < number; i++)
	{
     
		int temp;
		cin >> temp;
		v.push_back(temp);
	}
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(), v.end()), v.end());
	for(int i = 1; i < v.size(); i++)
		if(v[i] - v[i - 1] == 1)
		{
     
			if(v[i + 1] - v[i] == 1)
			{
     
				flag++;
				cout << "YES" << endl;
				break;
			}
		}
		else if(v[i] - v[i - 1] > 1) continue;
	if(flag == 0) cout << "NO" << endl;
	for(int i = 0; i < v.size(); i++)
		cout << v[i] << " ";
	cout << endl;
	return 0;
}

你可能感兴趣的:(排序,删除重复元素,vector,set,算法)