[파이썬/Python] 백준 10814번 나이순 정렬

[파이썬/Python] 백준 10814번 나이순 정렬

10814번: 나이순 정렬

---

 

문제

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

입력

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

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

출력

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

예제 입력

3
21 Junkyu
21 Dohyun
20 Sunyoung

예제 출력

20 Sunyoung
21 Junkyu
21 Dohyun

풀이 1

import sys
input = sys.stdin.readline

# 사용자 수를 n에 저장
n = int(input())

# 사용자 정보를 저장할 리스트 생성
user = []

# n번 만큼 반복하며 사용자 정보를 입력받고 리스트에 추가
for _ in range(n):
    age, name = input().split()  # 나이와 이름을 공백으로 구분하여 입력
    user.append((age, name))  # 나이와 이름을 튜플로 묶어 리스트에 추가

# 나이를 기준으로 리스트를 정렬
# lambda 함수를 사용하여 리스트의 요소에서 나이를 추출하고 int형으로 변환
user = user.sorted(user, key=lambda x: int(x[0]))

# 정렬된 리스트를 출력
for age, name in user:
    print(age, name)

처음 문제를 풀 때 딕셔너리로 접근했으나, 사용자의 이름과 나이가 전부 같은 경우는 고려를 못했었다.

따라서 딕셔너리 보다는 튜플 리스트로 접근하는게 더 간편했다.

또한 정렬을 수행할 때 int 형변환을 하지 않는다면 문자열 10 의 1 이 9 보다 먼저 정렬되는 부분도 생각하지 못했었다.

 

이 코드의 시간 복잡도는 O(n log n)이다. 이는 sorted() 함수가 퀵 정렬 알고리즘을 사용하기 때문이다. 따라서, 입력의 크기 n에 대해 약 n log n번의 연산이 수행되며, 입력이 매우 큰 경우에도 빠르게 정렬할 수 있다.

풀이 2

import sys

# 입력에서 첫 번째 줄을 제외한 모든 줄을 읽어옴
name_list = sys.stdin.readlines()[1:]

# 나이를 기준으로 이름을 정렬
# lambda 함수를 사용하여 각 줄에서 나이를 추출하고 int형으로 변환
name_list.sort(key=lambda iD : int(iD.split()[0]))

# 정렬된 이름 리스트를 출력
# 리스트의 각 요소를 이어붙인 문자열을 출력
print("".join(name_list))

이 코드는 알고리즘의 동작 방식 자체는 같으나, 입출력면에서 실행 시간을 더 줄일 수 있는 방법이다.

코드가 매우 간결하고 깔끔하다.

참고자료

로그인

글 읽기 - 출력 형식 모두 맞게 나오는데 왜 틀린지 모르겠습니다..

글 읽기 - 어느 부분이 틀린건지 궁금합니다

[코딩 스터디] 파이썬 정렬 함수 sorted _ key = lambda