코드연습/BOJ
BOJ 10814 : 나이순 정렬
AI 로밧
2022. 4. 26. 14:51
728x90
문제
온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.
풀이
직전 문제와 동일하게 Nested List로 처리하면 금방 풀린다. 그러나 시간이 조금 필요했던 부분은 age를 string으로 받아서 ,int로 받았을 때와 결과가 다르게 나온다는 것. 이해하고 처리하니 금방 해결됐다.
import sys
if __name__ == '__main__':
N = int(input())
data_list = {}
for _ in range(N):
age, name = sys.stdin.readline().split()
if int(age) in data_list:
data_list[int(age)].append(name)
else:
data_list[int(age)] = [name]
data_list = dict(sorted(data_list.items(), key = lambda item: item[0]))
for k,v in data_list.items():
for d in v:
print(k, d)