蓝桥杯——真题训练之蚂蚁感冒

标题：蚂蚁感冒
    长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

import java.util.Scanner;

public class Main{

	static boolean[] dir;//true: 表示方向向右	false: 向左
	static int[] dis;//绝对距离
	static boolean[] gm;//是否感冒

	public static void main(String[] args) {
		// TODO Auto-generated method stub

		Scanner scanner = new Scanner(System.in);
		int n = Integer.parseInt(scanner.nextLine());
		dir = new boolean[n];
		dis = new int[n];
		gm = new boolean[n];

		String[] sarr = scanner.nextLine().split("\\s+");
		if (Integer.parseInt(sarr[0]) > 0) {
			dir[0] = true;
			dis[0] = Integer.parseInt(sarr[0]);
			gm[0] = true;
		}
		else
		{
			dir[0] = false;
			dis[0] = -Integer.parseInt(sarr[0]);
			gm[0] = true;
		}
		for (int i = 1; i < n; i++) {
			int tmp = Integer.parseInt(sarr[i]);
			if (tmp > 0) {
				dir[i] = true;
				dis[i] = tmp;
			} else {
				dir[i] = false;
				dis[i] = -tmp;
			}
		}
		
		cal();
		int count=0;
		for(boolean b:gm)
		{
			if(b)
			{
				count++;
			}
		}
		
		System.out.print(count);
	}

	static void cal() {
		for(int i=1;i<dis.length-1;i++)
		{
			for(int j=i+1;j<dis.length;j++)
			{
				if(Math.abs(dis[i])>Math.abs(dis[j]))
				{
					int tmp=dis[i];
					dis[i]=dis[j];
					dis[j]=tmp;
					
					boolean b=dir[i];
					dir[i]=dir[j];
					dir[j]=b;
				}
			}
		}
		
		/*for(int i:dis)
		{
			System.out.print(i+" ");
		}*/
		if (dir[0]) {
			int mark=0;
			for (int i = gm.length-1; i >0; i--) {
				if (dis[0] < dis[i]) {
					if (dir[i]) {

					} else {
						gm[i] = true;
						mark=1;
					}
				}
				else {
					if(dir[i]&&mark==1)
					{
						gm[i]=true;
					}
				}
			}
		}
		else {
			int mark=0;
			for (int i = 1; i <gm.length; i++) {
				if (dis[0] > dis[i]) {
					if (!dir[i]) {

					} else {
						gm[i] = true;
						mark=1;
					}
				}
				else {
					if(!dir[i]&&mark==1)
					{
						gm[i]=true;
					}
				}
			}
		}
	}
}