백준 10814번 파이썬 풀이: 나이순 정렬

백준 10814번 나이순 정렬

 

알고리즘 분류: 정렬

 

링크: https://www.acmicpc.net/problem/10814

10814번: 나이순 정렬

문제

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

입력

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

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

출력

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

 

 

문제 파악

 

이 문제는 나이 순으로 정렬을 하고 나이가 같을시 알파벳 순으로 정렬을 하는게 아닌

 

먼저 가입을 한 순서로 정렬을 해주어야 한다.

 

그렇기 때문에 가입한 순서를 따로 선언해줘서 이에 따른 정렬을 해주어야 한다.

 

 

코드

# 10814번

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

for i in range(n):
    user = input().split()
    user.append(i)
    user_list.append(user)

sorted_list = sorted(user_list, key=lambda x : (int(x[0]), x[2]))
for i in range(len(sorted_list)):
    answer = " ".join(sorted_list[i][:2])
    print(answer)

처음 반복문을 도는동안 나이와 이름을 입력값으로 받아서 리스트를 만드는데

 

이 때 받는 순서대로 순서를 user 리스트에 추가해준다.

 

이렇게 만들어진 리스트를 user_list라는 배열에 다시 추가해줘서 이중배열을 만든다.

 

user_list를 정렬해주는데 먼저 나이순으로 정렬해주고 그 다음은 가입한 순서로 정렬해준다.

 

이 때 나이순으로 정렬할시에 int형으로 정렬을 하지않고 그냥 str형 그대로 정렬을 해버리면

 

예를들어 123이 25보다 앞으로 와버리는 오류를 범할 수 있으니 주의해야 한다.

 

그렇게 리스트를 정렬했으면 다시 문자열로 합쳐주면 되는데 출력을 할 때

 

순서는 출력을 안해도 되기 때문에 슬라이싱해서 없애주고 문자열을 출력해주면 된다.