蓝桥杯 历届试题 蚂蚁感冒

问题描述
　　长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

假如感冒蚂蚁向右走，那么碰到所有想左走的都会被感染，而感染后的蚂蚁必定是向左走的，那么他会把左边向右走的都感染了。向左走的也是这样。
ans=左边向右走的+右边向左走的+1（本身感冒）；
有特殊情况，第一只感染的向右走，右边的都在向右走，那么速度一样的话它不会感染其他所有，所以 ans = 1，相反也是。

import java.util.Scanner;

public class Main {

    public static void main(String args[]) {

        Scanner input = new Scanner(System.in);
        int n;
        int a[] = new int[55];
        n = input.nextInt();
        for (int i = 0; i < n; i++) {
            a[i] = input.nextInt();
        }
        int m=a[0];
        int tmp;
        for(int i=0;i<n-1;i++)
        {
            for(int j=0;j<n-1-i;j++)
            {
                int p=Math.abs(a[j]);
                int q=Math.abs(a[j+1]);
                if(q<p)
                {
                    tmp=a[j];
                    a[j]=a[j+1];
                    a[j+1]=tmp;
                }
            }
        }
        int t = 0;
        for(int i=0;i<n;i++)
        {
            if(m==a[i])
            {
                t=i;
            }

        }
        int left=0,right=0;
        for(int i=0;i<t;i++)
        {
            if(a[i]>0)
                left++;
        }
        for(int i=t+1;i<n;i++)
        {
            if(a[i]<0)
                right++;
        }
        int ans=0;
        if(a[t]>0&&right==0||a[t]<0&&left==0)
        {
             ans=1;
        }
        else
        {
            ans=left+right+1;
        }
        System.out.println(ans);
    }
}

其实这个是别人的思路，我本来是想简单模拟，每次传染后都重新改变数组，样例可以过，但是提交之后结果错误，现在仍不知道哪里错了，希望明白的朋友可以指点一下。

下面是错误代码：

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static void main(String args[]) {

        Scanner input = new Scanner(System.in);
        int n;
        int a[] = new int[55];
        int flag[] = new int[55];
        n = input.nextInt();
        for (int i = 0; i < n; i++) {
            a[i] = input.nextInt();
        }
        int m=a[0];
        int tmp;
        for(int i=0;i<n-1;i++)
        {
            for(int j=0;j<n-1-i;j++)
            {
                int p=Math.abs(a[j]);
                int q=Math.abs(a[j+1]);
                if(q<p)
                {
                    tmp=a[j];
                    a[j]=a[j+1];
                    a[j+1]=tmp;
                }
            }
        }

        for(int i=0;i<n;i++)
        {
            if(m==a[i])
            {
                flag[i]=1;
            }

        }
        while (true) {
            int s=1;
            for (int i = 0; i<n; i++) 
            {
                for(int j=i+1;j<n;j++)
                {
                    if (a[i] > 0 && a[j] < 0) {
                        if(flag[i]+flag[j]==1)
                        {
                            s=0;
                            flag[i] = 1;
                            flag[j] = 1;
                            a[i] = 0-a[i];
                            a[j] = 0-a[j];
                        }   
                    }
                }
            }
            if(s==1)
                break;

        }
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (flag[i] == 1) {
                count++;
            }
        }
        System.out.println(count);
    }
}