백준 10814: 나이순 정렬

백준 문제 링크

문제

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

입력

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

출력

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

---

아이디어

js에서는 정렬을 할 때 애초에 기준을 사용자가 직접해야해서 간단한 문제이다.
그런데 python에서는 그런 방법이 있는지 몰라서 검색해봤는데 있어서 js스럽게 문제를 풀이했다.

---

제출 코드

N = int(input())
member_list = []
for i in range(N):
    age, name = input().split()
    age = int(age)
    member_list.append({"age": age, "name": name})
sorted_member = sorted(member_list, key=lambda member: member["age"])
for member in sorted_member:
    print(member["age"], member["name"])

---

다른 개발자의 코드를 참고하여 리팩토링

일단 member_list를 만드는 방법 자체를 sys.stdin.readlines로 하면
굳이 객체를 만들지 않아도 한번에 배열도 된다는 것을 알았다.

import sys

name_list = sys.stdin.readlines()[1:]
name_list.sort(key=lambda member: int(member.split()[0]))

print("".join(name_list))

기존버전과 리팩토링 버전

메모리와 시간 모두 압도적으로 차이가 난다.

sys.stdin.readlines라는걸 새로 배웠다!

앞으로 input말고 저걸 열심히 활용해봐야지.