蓝桥杯-蚂蚁感冒

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

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

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

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

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

【数据格式】

第一行输入一个整数n (1 < n < 50), 表示蚂蚁的总数。

接着的一行是n个用空格分开的整数 Xi (-100 < Xi < 100), Xi的绝对值，表示蚂蚁离开杆子左边端点的距离。正值表示头朝右，负值表示头朝左，数据中不会出现0值，也不会出现两只蚂蚁占用同一位置。其中，第一个数据代表的蚂蚁感冒了。

要求输出1个整数，表示最后感冒蚂蚁的数目。

例如，输入：
3
5 -2 8
程序应输出：
1

再例如，输入：
5
-10 8 -20 12 25
程序应输出：
3

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 1000ms
先判断蚂蚁的方向，再算和蚂蚁相反方向有几只蚂蚁，然后蚂蚁掉头，算另一个相反方向上有几只蚂蚁，然后感冒的总数就等于两次求得和。水题。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std; 
int main(){
    int num[51],f[51];
    int n,i,j,cnt;
    while(scanf("%d",&n)!=EOF){
        for(i=0;i<n;i++){
            scanf("%d",&num[i]);

        }
        cnt=0;
        memset(f,0,sizeof(int)*n);
        if(num[0]<0){
            for(i=1;i<n;i++){
                if(abs(num[i])<abs(num[0])&&num[i]>0&&!f[i]){
                    cnt++;
                    f[i]=1;
                }
                if(cnt>0){
                    for(j=1;j<n;j++)
                    if(abs(num[j])>abs(num[0])&&num[j]<0&&!f[j]){
                        cnt++;
                        f[j]=1;
                    }
                }
            }
            cnt++;
        }
        else{
            for(i=1;i<n;i++){
                if(abs(num[i])>abs(num[0])&&num[i]<0&&!f[i]){
                    cnt++;
                    f[i]=1;
                }
                if(cnt>0){
                    for(j=1;j<n;j++)
                    if(abs(num[j])<abs(num[0])&&num[j]>0&&!f[j]){
                        cnt++;
                        f[j]=1;
                    }

                }

            }
            cnt++;          
        }
        printf("%d\n",cnt);
    }
    return 0;
}