[백준 알고리즘] 10814번: 나이순 정렬 (파이썬 / Python)

문제

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

입력

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

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

출력

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

예제 입력

3
21 Junkyu
21 Dohyun
20 Sunyoung

예제 출력

20 Sunyoung
21 Junkyu
21 Dohyun

 

---

파이썬 문법에 대한 이해도가 부족해서 복잡한 방법으로 풀다가 시간초과가 났다.

코드는 다음과 같다.

 

시간초과 코드

import sys

n = int(sys.stdin.readline())
a = []
for i in range(n):
    a.append(sys.stdin.readline().split())
b = a.copy()
a = sorted(a)
for i in range(len(a)):
    if i == 0: continue
    if int(a[i-1][0]) == int(a[i][0]) and b.index(a[i]) < b.index(a[i-1]):
        a[i] ,a[i-1] = a[i-1], a[i]

for i in range(len(a)):
    print(a[i][0], a[i][1])

시간초과가 났지만 코드 설명을 적어보자면,

1. 나이, 이름을 입력받아 a라는 리스트를 만든다.
2. a리스트를 copy하여 b리스트를 만든다.
3. a리스트를 sort한다. (이때 a리스트는 나이순 + 이름순(알파벳순)으로 정렬된다)
4. a리스트를 돌며, a의 전 index와 현재 index의 이름이 같을 때, b리스트에서는 순서가 반대인 경우 swap한다.
5. 출력

index함수를 사용하는 부분에서 시간초과가 생긴 것 같다.

 

정답 코드

import sys

n = int(sys.stdin.readline())
a = []
for i in range(n):
    age, name = sys.stdin.readline().split()
    age = int(age)
    a.append([age, name])

a.sort(key = lambda x : x[0])

for i in range(len(a)):
    print(a[i][0], a[i][1])

수정한 코드이다.

변경된 부분은 입력받은 후 나이를 int로 변경해주었고, sort를 key = lambda를 사용해 나이 하나만의 기준으로 정렬하도록 간단하게 구현할 수 있었다.

아직 너무 부족한 것 같다!

---

 

 

10814번: 나이순 정렬