[蓝桥杯历届试题-蚂蚁感冒] 数学解法

解题思路:

以感染蚂蚁为中心，计算向感染蚂蚁靠近的蚂蚁的数量。

特判左右两边有一边蚂蚁数量为0的情况。


参考代码:

#include <bits/stdc++.h>

using namespace std;

int main(void) {

    int n, a, l = 0, r = 0;

    cin >> n;

    cin >> a;

    int fa = abs(a);

    for (int i = 1; i < n; i++) {

        int b;

        cin >> b;

        if (b > 0 && abs(b) < fa) l++;

        if (b < 0 && abs(b) > fa) r++;

    }

    if ((a > 0 && r == 0) || (a < 0 && l == 0)) puts("1");

    else cout << (l + r + 1) << endl;

    return 0;

}