AcWing 1211. 蚂蚁感冒 解题思路及代码

先贴个题目：

以及原题链接：1211. 蚂蚁感冒 - AcWing题库https://www.acwing.com/problem/content/1213/

这题也是完全的数学题，蚂蚁速度相同，相撞后直接掉头，这样很难想具体的运动情况，但注意到速度相同，那不妨看成蚂蚁就是穿过并且感染，这就解决问题了。然后开始分类讨论分析什么样的蚂蚁会被传染，以初始蚂蚁向右为例，如果在初始蚂蚁的右侧，向右爬，永远都撞不到，就不会传染，如果是在右边，向左爬，必定撞上，如果是在左边，向左爬，必定撞不到，如果在左边向右爬，那么会不会被传染取决与右边有没有蚂蚁向左爬，分类讨论之后，用代码实现即可。

#include<iostream>
#include<cmath>
using namespace std;

const int N = 50;
int a[N];

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; ++i)
        cin >> a[i];
    int suml=0, sumr=0,sum=1;
    for (int i = 1; i < n;++i){
        if(a[0]>0){
            if(abs(a[0])>abs(a[i])&&a[i]>0)
                suml++;
            else if(abs(a[0])<abs(a[i])&&a[i]<0)
                sumr++;
        }  
        else{
            if (abs(a[i]) > abs(a[0]) && a[i]<0)
                suml++; 
            else if (abs(a[i]) < abs(a[0]) && a[i]>0)
                sumr++;     
        }
    }          
    if(sumr)
        sum+=suml+sumr;
    cout << sum;
    return 0;
}

 这里偷了个小懒，本来if（sumr）要写成两种情况的，我直接在a[0] <0的情况下吧suml和sumr的概念反了一下。

by————2024.2.12刷题记录