컴공생 누르지 마세요! 컴공생 울어요.

[정렬] 정렬 (5) 실전문제 - 성적이 낮은 순서로 학생 출력하기 본문

STUDY/알고리즘

[정렬] 정렬 (5) 실전문제 - 성적이 낮은 순서로 학생 출력하기

당도최고치악산멜론 2023. 3. 15. 11:29

이코테 p.180 실전문제 - 성적이 낮은 순서로 학생 출력하기

N명의 학생 정보가 있다. 학생 정보는 학생의 이름과 학생의 성적으로 구분된다.

각 학생의 이름과 성적 정보가 주어졌을 때, 성적이 낮은 순서대로 학생의 이름을 출력하시오.

입력 조건

1) 첫 번 째 줄에 학생의 수 N (1 <= N <= 100,000)

2) 두 번째 줄부터 N + 1번째 줄에는 학생의 이름을 나타내는 문자열 A와 학생의 성적을 나타내는 점수 B (A의 길이와 B는 100 이하의 자연수)

출력 조건

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

입력 예시

3
홍길동 95
이순신 77
이순실 77

출력 예시

이순신 이순실 홍길동

내 소스코드

  • 파이썬의 기본 정렬 라이브러리인 sorted() 함수 사용
  • key 매개변수에 lambda 식을 넣어 성적을 기준으로 오름차순 정렬
# 입력 받기
n = int(input())
array = []
for _ in range(n):
  array.append(list(input().split()))

# 성적을 key 값으로 오름차순 정렬
result = sorted(array, key = lambda x : x[1])

for i in range(n):
  print(result[i][0], end = ' ')

교재 소스코드

  • 학생 정보가 최대 100,000개까지 입력될 수 있으므로 최악의 경우 O(NlogN)을 보장하는 알고리즘을 이용하거나, O(N)을 보장하는 계수 정렬 이용
  • 학생 정보를 (이름, 점수) 형태의 튜플로 묶은 뒤 점수를 기준으로 정렬 수행
  • 파이썬 기본 정렬 라이브러리 사용
# 입력 받기
n = int(input())
array = []
for _ in range(n):
  input_data = input().split()
  array.append((input_data[0], int(input_data[1])))

# 성적을 key 값으로 오름차순 정렬
array = sorted(array, key = lambda student : student[1])

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