历届试题 蚂蚁感冒

import java.util.Scanner;

public class Main { 

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        int[] Loction = new int[N+1];
        int[] direction = new int[N+1];
        for ( int i = 1 ; i <= N ; i++){
            Loction[i] = in.nextInt();
            direction[i] = Loction[i] < 0 ?0:1;
        }
        int ill_direction = (Loction[1] < 0)?0:1;
        int right = 0;
        int left = 0;
        int x = 0;
        int y = 0;
        for ( int i = 2 ; i <= N ; i++){
            //病蚂蚁向右走，蚂蚁i向左走，并且蚂蚁i在病蚂蚁的右边
            if ( Math.abs(Loction[1]) < Math.abs(Loction[i]) && Loction[1] > 0 && direction[i] == 0){
                right++;
            }
            //病蚂蚁向左走，蚂蚁i向右走，并且蚂蚁i在病蚂蚁的左边
            if ( Math.abs(Loction[1]) > Math.abs(Loction[i]) && Loction[1] < 0 && direction[i] == 1){
                left++;
            }
            if ( Math.abs(Loction[1]) > Math.abs(Loction[i]) && Loction[1] > 0 && direction[i] == 1){
                x++;
            }
            if ( Math.abs(Loction[1]) < Math.abs(Loction[i]) && Loction[1] < 0 && direction[i] == 0){
                y++;
            }
        }
        if ( ill_direction == 1){
            if ( right != 0){
                System.out.print(x+right+1);
            }else{
                System.out.print(1);
            }
        }else{
            if (left != 0){
                System.out.print(y+left+1);
            }else{
                System.out.print(1);
            }
        }
        in.close();
    }

}