第五届蓝桥杯省赛C++B组 蚂蚁感冒

标题：蚂蚁感冒
长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。 
每只蚂蚁都只能沿着杆子向前爬，速度是1厘米/秒。
当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。
这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。
请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

【数据格式】

第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。
接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。
要求输出1个整数，表示最后感冒蚂蚁的数目。

例如，输入：
3
5 -2 8
程序应输出：
1

再例如，输入：
5
-10 8 -20 12 25
程序应输出：
3

资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。
所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

思路：每只蚂蚁都是沿着杆子爬，速度是1cm/s，所以当蚂蚁碰面掉头的时候，我们就可以直接当做穿行，因为是其实是一样。那么会感染的蚂蚁就只有第一蚂蚁左侧向右走的和右侧向左走的，如果第一只蚂蚁向左走，它左侧的蚂蚁都向左走，那么就不会有蚂蚁被感染，如果第一只蚂蚁向右走，它右侧的蚂蚁都向右走，那么也不会被感染，排除这两种情况，其他都是左侧向右和右侧向左的会感染。参考：https://blog.csdn.net/y990041769/article/details/21941711

代码：

#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<map>  
#include<set>
using namespace std;

struct Ant
{
	int dis;
	int dir;
	int index;	
}ant[55];
int n;

int cmp(Ant a,Ant b)
{
	return a.dis<b.dis;
}
int main()
{
	cin>>n;
	int i,x;
	for(i=0;i<n;i++)
	{
		cin>>x;
		ant[i].dis=abs(x);
		ant[i].dir=x<0?-1:1;
		ant[i].index=i+1;
	}
	sort(ant,ant+n,cmp);
	int temp=0,left=0,right=0,ans=0;
	for(i=0;i<n;i++)
	{
		if(ant[i].index==1)
		{
			temp=i;
			break;
		}
		if(ant[i].dir==1)
			left++;
	}
	for(i=temp+1;i<n;i++)
	{
		if(ant[i].dir==-1)
			right++;		
	}
	if((ant[temp].dir==-1&&left==0)||(ant[temp].dir==1&&right==0))
		ans=1;
	else
		ans=1+left+right;
	cout<<ans<<endl;
	return 0;
}