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

柔光的暖阳◎ 2022-08-26 01:29 176阅读 0赞

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

##### 【数据格式】 #####

第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。

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

要求输出1个整数，表示最后感冒蚂蚁的数目。

##### 例如，输入： #####

5 -2 8

##### 程序应输出： #####

##### 再例如，输入： #####

\-10 8 -20 12 25

##### 程序应输出： #####

##### 资源约定： #####

峰值内存消耗 < 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;
    }