Java实现蚂蚁感冒问题

长 100 厘米的细长直杆子上有 n 只蚂蚁。

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

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

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

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

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

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

代码的核心是两个蚂蚁相遇可以看成是互相穿过,如果是传染的蚂蚁就变成两个传染体

import java.util.*;

public class Main{
    static int N = 105;
    static int[] st = new int[105]; 
    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int s = in.nextInt();
        int abss = Math.abs(s); //起始位置
        st[abss] = s;
        int res = 1;
        for(int i=1;i<n;i++){
            int x = in.nextInt();
            int absx = Math.abs(x);
            st[Math.abs(x)] = x;
            //后来想想也可以就用n做下标 这样更省空间
        }
        if(s>0){ //起始向右
            while(n-->0){
                for(int i=0;i<100;i++){
                if(abss<i && st[i]<0){
                    res++;
                    st[i]=0;
                    break;
                }
                if(abss>i && res>1 && st[i]>0){
                    res++;
                    st[i]=0;
                    break;
                }
                  
            }
            }
        }else{
            while(n-->0){
                for(int i=0;i<100;i++){
                if(abss>i && st[i]>0){
                    res++;
                    st[i]=0;
                    break;
                }
                if(abss<i && res>1 && st[i]<0){
                    res++;
                    st[i]=0;
                    break;
                }
                  
            }
            }
                
        }
        System.out.println(res);
        
    }
    
}