1211. 蚂蚁感冒

#include<iostream>
#include<math.h>
using namespace std;
int n, zuo, you, a[60],num;
int main()
{
    //不管他的碰头就掉头，因为碰头掉头可以看做是双方交换身份
    cin >> n;
    for (int i = 1; i <= n; i++) { cin >> a[i]; }
    for (int i = 2; i <= n; i++)
    {
        //zuo=在左侧并且向右走（就是zuo是必感染的）
        if (abs(a[i])<abs(a[1]) && a[i] > 0) { zuo++; }
        //you=在右侧并且向左走（就是you是必感染的）
        if (abs(a[1])<abs(a[i]) && a[i] < 0) { you++; }
    }
    //特殊情况：（只有一只感冒蚂蚁）
    //右侧没有向左走的蚂蚁
    if (a[1] > 0 && you == 0) { num = 1; }
    //左侧蚂蚁向右走的蚂蚁
    else if (a[1] < 0 && zuo == 0) { num = 1; }
    else { num =zuo + you + 1; }
    cout << num << endl;
    system("pause");
    return 0;
}