2014 第五届蓝桥杯软件本科A组预赛题解 编程之 蚂蚁感冒（nyoj990）

长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

解题思路

首先明白两只蚂蚁碰撞之后掉头和穿行过去是一样的，可以把穿行看做碰撞后掉头了，然后两个蚂蚁交换了，而是哪一只蚂蚁对结果不影响。

那么，假如第一只感冒蚂蚁向右走，那么碰到所有想左走的都会被感染，而感染后的蚂蚁必定是向左走的，那么他会把左边向右走的都感染了。

向左走的也是这样。

所以 ans = 左边向右走的  + 右边向左走的 + 1（本身）。

当然还有特殊情况，第一只感染的向右走，右边的都在向右走，那么速度一样的话它不会感染其他所有，所以 ans = 1，相反也是。

这是我个人思路，欢迎指正

nyoj题目：http://acm.nyist.net/JudgeOnline/problem.php?pid=990

我的代码：

#include <cstdio>
#include <algorithm>

using namespace std;
struct Node
{
    int x;
    int dis;
    int num;
};
Node a[100];
int comp(Node a1,Node a2)
{
    if(a1.x!=a2.x)
        return a1.x<a2.x;
}
int main()
{
    int n,x;
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)
        {
            scanf("%d",&x);
            a[i].dis=x<0?-1:1;
            a[i].x=abs(x);
            a[i].num=i+1;
        }
        sort(a,a+n,comp);  //排序
        int tmp=0,left=0,right=0,ans=0;
        for(int i=0;i<n;i++)  //左边向右走的
        {
            if(a[i].num==1)
            {
                tmp=i;break;
            }
            if(a[i].dis==1)
                left++;
        }
        for(int i=tmp+1;i<n;i++) //右边向左走的
        {
            if(a[i].dis==-1)
                right++;
        }
        if(a[tmp].dis==1 && right==0 || a[tmp].dis==-1 && left==0)
            ans=1;
        else
            ans=left+right+1;
        printf("%d\n",ans);
    }
    return 0;
}