华为od机试 Python 【选修课】

偏执的太偏执、 2024-03-08 07:49 114阅读 0赞

### 题目 ###

你得到了两个列表，每个列表里是某门课程的学生的成绩。每个学生有一个唯一的8位数字学号，你可以从学号中看出他的班级。现在，请帮我们找出这样的学生：他们两个课都修了，成绩都有。

输出规则：

如果没学生两门课都修了，就输出"NULL"。  
先按班级排序。小号班级的先输出。  
在每个班级里，先输出这个班级的编号（学号的前5位）。  
接着输出这个班级修了两门课的学生学号。先按这两门课成绩的和从高到低排序，成绩和相同的，学号小的先输出。学号之间用英文分号隔开。  
输入格式：

第一行：第一门课的学生成绩。学生之间用英文分号隔开，学号和成绩用英文逗号隔开。  
第二行：和第一行一样，但是是第二门课的学生成绩。  
注意事项：

学号：8位数字。结构如下：2位院系编号 + 2位入学年份 + 1位专业编号 + 3位班级内学号。  
成绩：0到100之间的整数。  
示例：

输入 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

输入 01201022,75;01202033,95;01202018,80;01203006,90;01202066,100  
01202008,70;01203102,85;01202111,80;01201021,75;01201100,88  
输出 NULL  
说明 没有同时选修了两门选修课的学生，输出NULL.

### 代码 ###

def find_common_students(scores1_str, scores2_str):
        def parse_scores_line(line):
            scores_dict = {
        }
            for item in line.split(";"):
                student, score = item.split(",")
                scores_dict[student] = int(score)
            return scores_dict
    
        scores1 = parse_scores_line(scores1_str)
        scores2 = parse_scores_line(scores2_str)
    
        # 找出同时选了两门课的学生，并计算他们的总分
        shared_students = {
        }
        for student, score1 in scores1.items():
            if student in scores2:
                total_score = score1 + scores2[student]
                shared_students[student] = total_score
    
        if not shared_students:
            return "NULL"
    
        # 按班级分组
        grouped_by_class = {
        }
        for student, total_score in shared_students.items():
            class_id = student[:5]
            if class_id not in grouped_by_class:
                grouped_by_class[class_id] = []
            grouped_by_class[class_id].append((student, total_score))
    
        # 排序输出
        output = []
        for class_id, students in sorted(grouped_by_class.items()):
            # 按照总分和学号排序
            students.sort(key=lambda x: (-x[1], x[0]))
            student_ids = ";".join([student[0] for student in students])
            output.append(class_id)
            output.append(student_ids)
    
        return "\n".join(output)
    
    scores1 = "01202021,75;01201033,95;01202008,80;01203006,90;01203088,100"
    scores2 = "01202008,70;01203088,85;01202111,80;01202021,75;01201100,88"
    print(find_common_students(scores1, scores2))
    
    scores1 = "01201022,75;01202033,95;01202018,80;01203006,90;01202066,100"
    scores2 = "01202008,70;01203102,85;01202111,80;01201021,75;01201100,88"
    print(find_common_students(scores1, scores2))