需求:
题目描述
幼儿园两个班的小朋友在排队时混在了一起,每位小朋友都知道自己是否与前面一位小朋友同班,请你帮忙把同班的小朋友找出来小朋友的编号是整数,与前一位小朋友同班用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/YScanner 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")) { //判断尾端是否为Ntemp = temp.equals(arr1) ? (arr2) : (arr1); }//添加到最后temp.offerLast(chs);}}}
效果: