성적이 낮은 순서로 학생 출력하기-[이것이 취업을 위한 코딩 테스트다]

📖문제

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

입력 조건

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

출력 조건

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

My answer

import sys
input= sys.stdin.readline

n=int(input())
array=[]

for i in range(n):
    name,score=input().split()
    array.append([name,int(score)])
array=sorted(array,key=lambda x:x[1])

for name in array:
    print(name[0],end=" ")

Another answer

# N 입력 받기
n = int(input())

# N명의 학생 정보를 입력 받아 리스트에 저장
array = []
for i 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=' ')

문자형과 정수형을 동시에 입력받으므로, 문자형으로 받아주고 배열에 넣을 때 정수형으로 변경해서 넣어준다음(숫자를 정렬하려면 반드시 문자가 아니라 숫자여야함) sorted 함수에서 lambda를 이용하여 점수를 기준으로 정렬해준다. 

숫자로 정렬하고 이름을 출력해야하기 때문에 위와 같이 lambda라는 조건을 이용한 정렬을 한 것.