21天学习挑战赛-基础题-逻辑-分班

​
​

活动地址：CSDN21天学习挑战赛

题目描述

幼儿园两个班的小朋友在排队时混在了一起，每位小朋友都知道自己是否与前面一位小朋友是否同班，请你帮忙把同班的小朋友找出来。
小朋友的编号为整数，与前一位小朋友同班用Y表示，不同班用N表示。

题目要求

输入为：空格分开的小朋友编号和是否同班标志。
比如：6/N 2/Y 3/N 4/Y，表示共4位小朋友，2和6同班，3和2不同班，4和3同班。
其中，小朋友总数不超过999，每个小朋友编号大于0，小于等于999。不考虑输入格式错误问题。
输出为：

• 两行，每一行记录一个班小朋友的编号，编号用空格分开。且：
• 编号需要按照大小升序排列，分班记录中第一个编号小的排在第一行。
• 若只有一个班的小朋友，第二行为空行。
• 若输入不符合要求，则直接输出字符串ERROR。

示例

输入

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

代码解析

图解

题目要求很明确，最后输出要求是在分班完的基础上对输出进行排序以及大小判断处理；下面代码不包含判断格式的要求，可以用正则表达式判断。
代码如下：

import java.util.Arrays;
import java.util.Scanner;

public class Day01DivClass {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        //将输入的内容以数组形式存储
        String[] l = in.nextLine().split(" ");
        //初始化1班2班
        int[] clazz1 = new int[l.length];
        int[] clazz2 = new int[l.length];
        //默认第一名同学为A班
        clazz1[0] = Integer.parseInt(l[0].substring(0,l[0].lastIndexOf("/")));
        //设置标志位判断
        boolean isSameClass = true;
        //对数组遍历判断分班
        for (int i = 1; i < l.length; i++) {
            String c = l[i].substring(l[i].lastIndexOf("/")+1);
            if(isSameClass){   //
                if(c.equals("Y")){
                    clazz1[i] = Integer.parseInt(l[i].substring(0,l[i].lastIndexOf("/")));
                }else {
                    clazz2[i] = Integer.parseInt(l[i].substring(0,l[i].lastIndexOf("/")));
                    isSameClass = false;
                }
            }else {
                if(c.equals("Y")){
                    clazz2[i] = Integer.parseInt(l[i].substring(0,l[i].lastIndexOf("/")));
                }else {
                    clazz1[i] = Integer.parseInt(l[i].substring(0,l[i].lastIndexOf("/")));
                    isSameClass = true;
                }
            }
        }
        //编号需要按照大小升序排列
        Arrays.sort(clazz1);
        Arrays.sort(clazz2);
        String clazzA = "";
        String clazzB = "";
        //编号>0且<=999
        for (int i : clazz1) {
            if(i!=0){
                clazzA =clazzA.length()>0?clazzA+" "+i:clazzA+i;
            }
        }
        for (int i : clazz2) {
            if(i!=0){
                clazzB = clazzB.length()>0?clazzB+" "+i:clazzB+i;
            }
        }
        //分班记录中第一个编号小的排在第一行
        if(clazzA.length()>0&&clazzB.length()>0){
            if(Integer.parseInt(clazzA.split(" ")[0])<Integer.parseInt(clazzA.split(" ")[0])){
                System.out.println(clazzA);
                System.out.println(clazzB);
            }else {
                System.out.println(clazzB);
                System.out.println(clazzA);
            }
        }
    }

}

​