Course List for Student

def solve(nk, courses_data, queries_data):
    n, k = map(int, str(nk).split())
    student_courses = {}
    for rec in str(courses_data).strip().split(';'):
        parts = rec.split(':')
        course = parts[0]
        students = parts[1].split(',') if len(parts) > 1 else []
        for stu in students:
            stu = stu.strip()
            if stu:
                if stu not in student_courses:
                    student_courses[stu] = []
                student_courses[stu].append(course)
    for stu in str(queries_data).strip().split(','):
        stu = stu.strip()
        courses = sorted(student_courses.get(stu, []))
        print(f'{stu}: {" ".join(map(str, courses))}')