蚂蚁感冒

【题目描述】
长 100 厘米的细长直杆子上有 n 只蚂蚁。

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

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

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

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

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

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

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

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

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

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

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

【题目分析】
《1》题目提到两只蚂蚁相碰时会掉头，并且会感染病毒，此时可以看作两只蚂蚁穿越过去，往右边走的继续往右边走，往左边走的继续往左边走，不同的是如果有病毒会感染病毒
《2》我们以第一只带病毒的蚂蚁为原点
如果这只带病毒的蚂蚁向右边走的话，我们知道
1.右边的向左走的蚂蚁一定会被感染
2.如果存在右边的向左走的蚂蚁，那么左边向右边走的蚂蚁也会被感染
如果不存在的话，那么左边向右边走的蚂蚁则不会被感染
如果这只带病毒的蚂蚁向左边走的话，我们知道
1.左边的向右走的蚂蚁一定会被感染
2.如果存在左边向右走的蚂蚁，那么右边向左走的蚂蚁一定被感染
如果不存在的话，那么右边向左走的蚂蚁则不会被感染

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 51;
int w[N];
int n;
int ans;

int main()
{
    cin >> n;
    for(int i = 1;i <= n;i++) cin >> w[i];

    int right = 0; //right表示右边向左走的蚂蚁
    int left = 0;  //left表示左边向右走的蚂蚁
    for(int i = 2;i <= n;i++)
    {
        if(abs(w[i]) < abs(w[1]) && w[i] > 0)  left++ ;
        if(abs(w[i]) > abs(w[1]) && w[i] < 0)   right++ ;
    }
    if(w[1] > 0)
    {
        if(right == 0) ans = 1; //左边向右边走的蚂蚁不存在，那么右边向左走的蚂蚁不会被感染
        else ans = right + left + 1;//存在则右边向左走的蚂蚁会被感染，相加并且加上初始生病的蚂蚁
    }

    else
    {
        if(left == 0) ans = 1; //与上情况性质相同
        else ans = right + left + 1;
    }

    printf("%d",ans);
}