📖문제

N명의 학생 정보가 있다. 학생 정보는 학생의 이름과 학생의 성적으로 구분된다. 각 학생의 이름과 성적 정보가 주어졌을 때 성적이 낮은 순서대로 학생의 이름을 출력하는 프로그램을 작성하시오.

입력 조건

• 첫 번째 줄에 학생의 수 N이 입력된다.(1 <= N <= 100,000)
• 두 번째 줄부터 N + 1번째 줄에는 학생의 이름을 나타내는 문자열 A와 학생의 성적을 나타내는 정수 B가 공백으로 구분되어 입력된다. 문자열 A의 길이와 학생의 성적은 100 이하의 자연수이다.

출력 조건

• 모든 학생의 이름을 성적이 낮은 순서대로 출력한다. 성적이 동일한 학생들의 순서는 자유롭게 출력해도 괜찮다.

📝풀이과정

• 이 문제는 계수 정렬을 이용해서 풀으라고 준 문제라고 생각하여 계수 정렬으로 풀었다.
  계수 정렬은 데이터의 크기가 한정되어있을 때 쓰기에 적합한 알고리즘인데, 이 문제는 정렬할 데이터의 최솟값과 최댓값의 차이가 100이기 때문이다.

1. 점수 리스트를 2차원 리스트로 생성한다. (scores)
   이 리스트에는 해당 점수(인덱스)를 받은 학생들의 이름이 들어갈 것이다.
2. name과 score을 공백으로 구분하여 입력받은 뒤에 score는 정수형으로 변환해준다.
3. 리스트 scores에서 인덱스가 score인 곳에 방금 입력 받은 name을 추가한다.
4. 2~3번을 n번 반복

⌨코드1 (내 코드)

n = int(input())

scores = [[] for _ in range(101)]

for i in range(n):
  name, score = input().split()
  score = int(score)
  scores[score].append(name)

for i in range(101):
  for n in scores[i]:
    print(n, end = ' ')

⌨코드2 (책 예제 코드)

• 책에서는 계수 정렬로 풀지 않고 리스트에 학생의 이름과 점수를 튜플로 만들어 리스트에 저장한 후 파이썬의 기본 정렬 라이브러리(sorted)를 사용해서 문제를 풀었다.
• 기본 정렬 라이브러리에서 람다 함수로 key를 만들어 쓰는 방법을 알아두면 좋을 것 같아서 책 예제 코드도 써보았다.

n = int(input())

array = []
for i in range(n):
    input_data = input().split()
    array.append((input_data[0], int(input_data[1])))
    
array = sorted(array, key = lambda student: student[1])

for student in array:
    print(student[0], end = ' ')