[PART2] 6-2(정렬): 성적이 낮은 순서로 학생 출력하기

코뉴·2021년 1월 23일

이코테 정렬

이코테: 문제풀이

목록 보기

19/28

💥이코테 실전문제 뽀개기💥

💻 6-1 위에서 아래로

난이도🖤🤍🤍 | 풀이시간 20분 | 제한시간 1초 | 메모리제한 128MB | 기출 D 기업 프로그래밍 콘테스트 예선

📌2021/01/23 작성 코드

n = int(input())
grades = []
for _ in range(n):
    data = input().split()
    grades.append([data[0], int(data[1])])

# key에 lambda식 사용해 봄!
grades.sort(key=lambda grade : grade[1])
for grade in grades:
    print(grade[0], end=' ')

💭 아이디어

학생 정보에서 성적을 기준으로 sort 해야하므로 key 값으로 lambda 식을 써서 정렬해 보았다.

🤓 문제 해설

학생의 정보가 최대 100,000개까지 입력될 수 있으므로 최악의 경우 O(NlogN)을 보장하는 알고리즘을 이용하거나 O(N)을 보장하는 계수 정렬을 이용하면 된다.

입력되는 데이터는 학생의 이름과 점수지만 출력할 때는 이름만 출력하면 되므로 학생 정보를 (점수, 이름)으로 묶은 뒤 점수를 기준으로 정렬을 수행한다.

파이썬의 기본 정렬 라이브러리를 사용하는 것이 효과적.
튜플을 이용했다.

➕ 튜플을 사용하는 이유는 리스트에 비해서 조금 더 공간 효율적이기 때문이다. 리스트는 요소를 추가하는 동작을 빠르게 수행할 수 있도록 더 많은 공간을 저장해둔다. (출처: https://edykim.com/ko/post/python-list-vs.-tuple/)

🤓 답안 예시

# 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=' ')