[算法题] 蚂蚁感冒

感谢观看，注释已尽量写清楚，谢谢

---

1. 题目

长 100 厘米的细长直杆子上有 n 只蚂蚁。

它们的头有的朝左，有的朝右。

每只蚂蚁都只能沿着杆子向前爬，速度是 1厘米/秒。

当两只蚂蚁碰面时，它们会同时掉头往相反的方向爬行。

这些蚂蚁中，有 1 只蚂蚁感冒了。

并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。

请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。

输入格式

第一行输入一个整数 n, 表示蚂蚁的总数。

接着的一行是 n 个用空格分开的整数 Xi, Xi 的绝对值表示蚂蚁离开杆子左边端点的距离。

正值表示头朝右，负值表示头朝左，数据中不会出现 00 值，也不会出现两只蚂蚁占用同一位置。

其中，第一个数据代表的蚂蚁感冒了。

输出格式

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

数据范围

1<n<50
0<|Xi|<100

输入样例1：

3
5 -2 8

输出样例1：

1

输入样例2：

5
-10 8 -20 12 25

输出样例2：

3

---

2. 思路

1.使用等价替换的思想：俩个蚂蚁相逢掉头 等价于 俩个蚂蚁相互穿过，
因为蚂蚁的速度相同，一只蚂蚁感染了，另一只蚂蚁也一定感染了，反之。
所以使用等价替换的思想化简题目意思

2.假设a[0]向左走，
那么a[0]左边往右走的一定会被感染；
左边往左走的一定不会被感染；
右边往右走的一定不会被感染；
右边往左走的有俩种情况：
1).a[0]左边往右走的数量==0，那么右边往左走的一定不会被感染；
2).a[0]左边往右走的数量>0 ，那么右边往左走的一定会被感染；

a[0]往左右的情况类似

---

3. 代码

#include<iostream>
#include<algorithm>
using namespace std;
int n;
int a[10000];

int left1,right1;//a[0]右边往左走，a[0]左边往右走的数量

//打印数据，调试用
void print()
{
    for(int i=0;i<n;i++)
        cout<<a[i]<<" ";
    cout<<endl<<left1<<" "<<right1<<endl;
    
}
int main()
{
   cin>>n;
   for(int i=0;i<n;i++) cin>>a[i];
    
    //先统计相对于中间值，即a[0],在a[0]左边往右走，和左边往左走的蚂蚁数量，其他俩种情况不可能感染，只有这种情况会感染
    for(int i=1;i<n;i++) 
    {
        if(abs(a[i])>abs(a[0])&&a[i]<0) left1++;
        else if(abs(a[i])<abs(a[0])&&a[i]>0) right1++;
        //print();
    }
    
    /*
    统计这俩种情况的结果
    1.a[0]左边往右走的数量==0，那么右边往左走的一定不会被感染；
    2.a[0]左边往右走的数量>0 ，那么右边往左走的一定会被感染；
    */
    if((a[0]<0&&right1==0)||(a[0]>0&&left1==0)) cout<<1;
    else if((a[0]>0&&left1>0)||(a[0]<0&&right1>0)) cout<<left1+right1+1;
    
    
    return 0;
}

---

4.附赠题目编译环境

[点我跳转~](1211. 蚂蚁感冒 - AcWing题库)

---

[神秘传送门😎](Hexo (meiyingg.github.io))

---

								难得写个笔记，点个赞再走呗~
                            			 肥肠感谢！
                               				 😁