问题 1454: [蓝桥杯][历届试题]蚂蚁感冒

题目描述
长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左，有的朝右。

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

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

这些蚂蚁中，有1只蚂蚁感冒了。并且在和其它蚂蚁碰面时，会把感冒传染给碰到的蚂蚁。

请你计算，当所有蚂蚁都爬离杆子时，有多少只蚂蚁患上了感冒。
输入
第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。

接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数 据代表的蚂蚁感冒了。
输出
要求输出1个整数，表示最后感冒蚂蚁的数目。
样例输入
5
-10 8 -20 12 25
样例输出
3

暴力 超时
发现一个规律
既然是一个会传染 那么 肯定是2边向中间跑的才会 传染
如图所示 红点为传染源 蓝点为其他正常蚂蚁

那么不管 传染源初始位置是向左还是向右， 那么在（存在有向右侧行驶的蚂蚁的情况下）传染源左侧的蚂蚁如果向右走就会传染， 右侧同理

所以 只要计算 左侧向右走的 和 右侧向左走的蚂蚁数量加1 即为结果

但有一种特殊情况
即 左侧都向左走 右侧都向右走 ，那么结果 即只有 传染源蚂蚁 一个

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include <bits/stdc++.h>
using namespace std;
bool cmp(int x, int y)
{
    return abs(x)<abs(y);
}
int main()
{
    int n;
    int a[1003];
    scanf("%d",&n);
    for (int i=0; i<n; i++)
    {
        scanf("%d",&a[i]);
    }
    int s=a[0];
    sort(a, a+n, cmp);
    int t;
    for (int i=0; i<n; i++)
    {
        if (a[i]==s)
        {
            t = i;
            break;
        }
    }
    int num1=0, num2=0;
    for (int i=t-1; i>=0; i--)
    {
        if (a[i]>0)
        {
            num1++;
        }
    }
    for (int i=t+1; i<n; i++)
    {
        if (a[i]<0)
        {
            num2++;
        }
    }
    if (a[t]<0&&num1==0||a[t]>0&&num2==0)
        printf ("1\n");
    else
        printf ("%d\n",num1+num2+1);
    return 0;
}