Java 华为真题-小朋友分班

需求： 

题目描述

幼儿园两个班的小朋友在排队时混在了一起，每位小朋友都知道自己是否与前面一位小朋友同班，请你帮忙把同班的小朋友找出来小朋友的编号是整数，与前一位小朋友同班用Y表示，不同班用N表示学生序号范围(0，999]，如果输入不合法则打印ERROR。

输入描述

        输入为空格分开的小朋友编号和是否同班标志

输出描述

        输出为两行，每一行记录一个班小朋友的编号，编号用空格分开，且:

        1.编号需按照升序排列，分班记录中第一个编号小的排在第一行。

        2.若只有一个班的小朋友，第二行为空行。

输入	1/N 2/Y 3/N 4/Y
输出	1 2 3 4
说明	2的同班标记为Y，因此和1同班。 3的同班标记为N，因此和1、2不同班。 4的同班标记为Y，因此和3同班。 所以1、2同班，3、4同班，输出为 1 2 3 4

编码：

public class TestClass {
    public static void main(String[] args) {
        //1/N 2/Y 3/N 4/Y
        //6/N 2/Y 3/N 4/Y 5/Y 7/N 8/Y
        Scanner scanner = new Scanner(System.in);
        System.out.print("输入小朋友的编号：");
        String str = scanner.nextLine();
//        System.out.println(str);
        String[] childrens = str.split(" ");
//        System.out.println(Arrays.toString(childrens));

        //创建双端队列对象
        ArrayDeque<String> arr_one = new ArrayDeque<>();
        ArrayDeque<String> arr_two = new ArrayDeque<>();

        //调用方法
        getBranch(childrens, arr_one, arr_two);
        show(arr_one);
        show(arr_two);

    }

    /**
     * 显示数据
     *
     * @param arr
     */
    private static void show(ArrayDeque<String> arr) {
        //使用Lambda,分割之后，再进行排序。
        System.out.println(arr.stream().map(s -> s.split("/")[0]).sorted().collect(Collectors.joining(" ")).trim());
    }

    /**
     * 分班
     * @param childrens
     * @param arr1
     * @param arr2
     */
    private static void getBranch(String[] childrens, ArrayDeque<String> arr1, ArrayDeque<String> arr2) {
        ArrayDeque<String> temp = null;
        //循环
        for (String chs :
                childrens) {
            //判断当前对象是否为空
            if (Objects.isNull(temp)) {
                temp = arr1; //给当前队列赋值
            } else if (chs.endsWith("N")) { //判断尾端是否为N
                temp = temp.equals(arr1) ? (arr2) : (arr1);  
            }
            //添加到最后
            temp.offerLast(chs);

        }

    }

}

效果：