【华为OD题库-005】选修课-Java

题目

现有两门选修课，每门选修课都有一部分学生选修，每个学生都有选修课的成绩，需要你找出同时选修了两门选修课的学生,先按照班级进行划分，班级编号小的先输出，每个班级按照两门选修课成绩和的降序排序，成绩相同时按照学生的学号升序排序。
输入描述
第一行为第—门选修课学生的成绩
第二行为第二门选修课学生的成绩，
每行数据中学生之间以英文分号分隔，每个学生的学号和成绩以英文逗号分隔，学生学号的格式为8位数字(2位院系编号+入学年份后2位+院系内部1位专业编号+所在班级3位学号)，学生成绩的取值范围为[0,100]之间的整数，两门选修课选修学生数的取值范围为[1-2000]之间的整数。
输出描述
同时选修了两门选修课的学生的学号，如果没有同时选修两门选修课的学生输出NULL，否则，先按照班级划分，班级编号小的先输出，每个班级先输出班级编号(学号前五位)，然后另起一行输出这个班级同时选修两门选修课的学生学号，学号按照要求排序(按照两门选修课成绩和的降序，成绩和相同时按照学号升序)，学生之间以英文分号分隔。
示例1
输入:
01202021,75;01201033,95,01202008,80;01203006,90;01203088,100
01202008,70;01203088,85;01202111,80;01202021,75;01201100,88
输出:
01202
01202008;01202021
01203
01203088
说明:
同时选修了两门选修课的学生01202021、01202008、01203088，这三个学生两门选修课的成绩和分别为150、150、185。01202021、01202008属于01202班的学生，按照成绩和降序，成绩相同时按学号升序输出的结果为：01202008:01202021；01203088属于01203班的学生，按照成绩和降序，成绩相同时按学号升序输出的结果为01203088，01202的班级编号小于01203的班级编号，需要先输出。
示例2
输入:
01201022,75;01202033,95;01202018,80;01203006,90;01202066,100
01202008,70;01203102,85;01202111,80;01201021,75;01201100,88
输出:
NULL
说明:
没有同时选修了两门选修课的学生，输出NULL。

思路

题目描述多，实际很简单。
用两个list分别记录两门课程对应的学生编号，两个list相交的部分就能得到“同时选修了两门选修课的学生的学号”。
关键在于排序，题目要求：先按班级升序，再按成绩降序，最后按学号升序排。班级和学号都可以从学生编号中提取，此处的成绩指的两门课的总成绩，所以可以使用一个map来根据学号查询总成绩。

题解

package hwod;

import java.util.*;

public class ElectiveCourse {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        Map<String, Integer> map = new HashMap<>(); //key为学号，value为总分
        List<String> firstIds = new ArrayList<>();
        List<String> secondIds = new ArrayList<>();
        for (int i = 0; i < 2; i++) {
            String line = sc.nextLine();
            for (String item : line.split(";")) {
                String sid = item.split(",")[0];
                String score = item.split(",")[1];
                map.put(sid, map.getOrDefault(sid, 0) + Integer.valueOf(score));
                if (i == 0) firstIds.add(sid);
                else secondIds.add(sid);
            }

        }
        List<String> out_ids = new ArrayList<>(); //同时选修了两门课程的学生学号
        for (int i = 0; i < firstIds.size(); i++) {
            if (secondIds.contains(firstIds.get(i))) {
                out_ids.add(firstIds.get(i));
            }
        }
        if (out_ids.size() == 0) {
            System.out.println("NULL");
            return;
        }
        //学号按照班级，成绩，学号排序
        Collections.sort(out_ids, (A, B) -> {
            if (!A.substring(0, 5).equals(B.substring(0, 5))) {
                return A.substring(0, 5).compareTo(B.substring(0, 5));
            }
            if (!map.get(A).equals(map.get(B))) {
                return -map.get(A).compareTo(map.get(B));
            }
            return A.compareTo(B);
        });

        //打印输出
        String classid = null;
        for (int i = 0; i < out_ids.size(); i++) {
            boolean isNewClass = false;
            if (classid == null || !classid.equals(out_ids.get(i).substring(0, 5))) {
                classid = out_ids.get(i).substring(0, 5);
                if(i!=0) System.out.println();
                System.out.println(classid);
                isNewClass = true;
            }
            if (isNewClass) System.out.print(out_ids.get(i));
            else System.out.print(";" + out_ids.get(i));


        }
    }

}