codeforces Div3 E. Boxers(签到题)

codeforces Div3 E. Boxers

题目

http://codeforces.com/problemset/problem/1203/E

题意

有n个人参加比赛，比赛要求参赛选手的体重都各不相同，每个人最多可以改变自己的体重1kg。
问最多能参赛多少名选手

题解

排序之后，优先判断-1kg能不能参加比赛，再判断改变0和1体重能不能参赛。

代码

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 150010;
int box[maxn],a[maxn];
int main()
{
	int n,ans = 0;
	memset(box,0,sizeof(box));
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
	sort(a,a+n);
	for(int i=0;i<n;i++)
	{
		if(a[i] != 1)
		{
			if(box[a[i]-1]==0)
			{
				box[a[i]-1]++;
				ans++;
			}
			else if(box[a[i]]==0)
			{
				box[a[i]]++;
				ans++;
			}
			else if(box[a[i]+1]==0)
			{
				box[a[i]+1]++;
				ans++;
			}
		}
		else
		{
			if(box[a[i]]==0)
			{
				box[a[i]]++;
				ans++;
			}
			else if(box[a[i]+1]==0)
			{
				box[a[i]+1]++;
				ans++;
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}