【华为OD机考B卷 | 100分】选修课（JAVA题解——也许是全网最详）

前言

本人是算法小白，甚至也没有做过Leetcode。所以，我相信【同为菜鸡的我更能理解作为菜鸡的你们的痛点】。

题干

1. 题目描述

现有两门选修课，每门选修课都有一部分学生选修，每个学生都有选修课的成绩，需要你找出同时选修了两门选修课的学生，先按照班级进行划分，班级编号小的先输出，每个班级按照两门选修课成绩和的降序排序，成绩相同时按照学生的学号升序排序。

2. 输入描述

第一行为第一门选修课学生的成绩；
第二行为第二门选修课学生的成绩。
每行数据中学生之间以英文分号分隔，每个学生的学号和成绩以英文逗号分隔，学生学号的格式为8位数字(2位院系编号+入学年份后2位+院系内部1位专业编号+所在班级3位学号)，学生成绩的取值范围为[0,100]之间的整数，两门选修课选修学生数的取值范围为[1-2000]之间的整数。

3. 输出描述

同时选修了两门选修课的学生的学号，如果没有同时选修两门选修课的学生输出NULL，否则，先按照班级划分，班级编号小的先输出，每个班级先输出班级编号（学号前五位），然后另起一行输出这个班级同时选修两门选修课的学生学号，学号按照要求排序（按照两门选修课成绩和的降序，成绩和相同时按照学号升序)，学生之间以英文分号分隔。

4. 示例

示例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。

解答

遇到的问题

说实在这一题遇到的问题难以描述。为啥？首先题目看似没什么难度，我直接用写业务的方式就能把代码写出来了，并且我百度找了好几个版本，也没看出来别人有啥特别亮眼的写法，都是在写业务代码而已。这恰恰就是问题所在。excuse me baby，我们不是在做算法题吗？不应该出现一点有逼格的解答吗？？
然后我看到部分专门做【华为OD】的大拿，虽然提示这一题要用【位运算】，但是我也没看到他的解答中用了位运算，我麻了。
经过多次百度，我看了一个我相对比较认可的答案，看他的意思，其实本题考察的是：考察的TreeMap或者TreeSet，或者Comparable相关的接口。

解题思路

根据条件，全文比较重要的线索是：选修2门课的学生。所以自然想到了Map数据结构。
结合排序条件，先班级，再分数，后学号。所以想到了TreeMap这个数据结构存储<班级，学生列表>的映射。它有以下特点：元素不重复；默认按升序排序。
使用TreeSet存储学生列表，并且TreeSet支持默认使用的排序接口Comparator

代码示例

public class OptionalCourse {
    static class Student {
         String id;
        int score;

        public Student(String id, int score) {
            this.id = id;
            this.score = score;
        }

        public String getId() {
            return id;
        }

        public int getScore() {
            return score;
        }
    }

       public static void main(String[] args) {

            // 读入数据
            Scanner scanner = new Scanner(System.in);
            String courseScore1 = scanner.nextLine();
            String courseScore2 = scanner.nextLine();

            String[] courseOne = courseScore1.split(";");
            String[] courseTwo = courseScore2.split(";");

            Map<String, Integer> studentMap = new HashMap<>();
            for (String t : courseTwo) {
                String[] student = t.split(",");
                String stuId = student[0];
                int stuScore = Integer.parseInt(student[1]);
                studentMap.put(stuId, stuScore);
            }

            Comparator<Student> stuComparator = Comparator.comparingInt(Student::getScore).thenComparing(Student::getId);
            TreeMap<String, TreeSet<Student>> map = new TreeMap<>();
            for (String s : courseOne) {
                String[] sStu = s.split(",");
                String sId = sStu[0];
                if (studentMap.containsKey(sId)) {
                    int sScore = Integer.parseInt(sStu[1]);
                    int tScore = studentMap.get(sId);
                    int totalScore = sScore + tScore;
                    String cls = sId.substring(0, 5);
                    Student student = new Student(sId, totalScore);
                    map.computeIfAbsent(cls, k -> new TreeSet<>(stuComparator)).add(student);
                }
            }

            if (map.isEmpty()) {
                System.out.println("NULL");
            } else {
                map.forEach((key, value) -> {
                    System.out.println(key);
                    String res = String.join(";", value.stream().map(Student::getId).toArray(String[]::new));
                    System.out.println(res);
                });
            }
    }
}