蓝桥杯2014——蚂蚁感冒

蓝桥杯2014——蚂蚁感冒

【题目】
长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

【注意点】
1、蚂蚁碰面后不要当做掉头，当做擦肩而过
2、忽略100cm长度
3、不要把目光局限于单个的蚂蚁，把蚂蚁分为 某个特殊蚂蚁左侧向右、向左爬行的蚂蚁 和 某个特殊蚂蚁右侧向右、向左的蚂蚁 四类来考虑即可

#include<iostream>
#include<algorithm>
using namespace std;
struct Ant
{
 int pos;
 int dir;
 int cool;
}ant[50];
bool cmp(Ant a1,Ant a2)
{
 return a1.pos<a2.pos;
}
int main()
{
 int n;
 cin>>n;
 int i,k;
 for(i=0;i<n;i++)
 {
  cin>>k;
  ant[i].pos=abs(k);
  ant[i].dir=(k<0)?-1:1;
  ant[i].cool=0;
 }
 ant[0].cool=1;
 sort(ant,ant+n,cmp);
 int lr=0,rl=0;
 for(i=0;i<n;i++)
 {
  if(ant[i].cool==1)
   break;
  if(ant[i].dir>0)
   lr++;
 }
 int ant_pos=i;
 for(i=ant_pos+1;i<n;i++)
  if(ant[i].dir<0)
   rl++;
 if((ant[ant_pos].dir>0&&rl==0)||(ant[ant_pos].dir<0&&lr==0))
  cout<<"1"<<endl;
 else
  cout<<rl+lr+1<<endl;
 return 0;
}