[정렬] 난이도1, D 기업 프로그래밍 콘테스트 예선 '성적이 낮은 순서로 학생 출력하기' (Python)

<문제>

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

 

<입력 조건>

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

<출력 조건>

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

---

이 문제는 파이썬 정렬 라이브러리로 풀 수 있었다.

 

<문제 풀이>

1. 이름과 성적이 짝을 이루기 때문에 dict로 값을 정렬
2. lambda 식을 이용해서 성적을 기준으로 내림차순 정렬
3. 정렬한 dict에서 이름만 추출

<코드>

'''입력예시
2
홍길동 95
이순신 77
'''

n = int(input())
student = {}

for i in range(n):
    name,grade = input().split()
    student[name] = int(grade)

student = dict(sorted(student.items(), key=lambda x:x[1]))

for i in student.keys():
    print(i,end=' ')

#220322
n = int(input())
answer = []

for i in range(n):
    answer.append(list(input().split()))

for i in sorted(answer,key=lambda x:x[1]):
    print(i[0],end=' ')

 

 

<고쳐야 할 점>

• dict를 이용한 lambda 반복 학습