(Python)백준 코딩테스트 연습 - 시리얼 번호(1431)

1. 문제

https://www.acmicpc.net/problem/1431

1431번: 시리얼 번호

2. 풀이

import sys
input = sys.stdin.readline
# n : 케이스 수
n = int(input())

# n_d : 케이스 저장할 딕셔너리
n_d = {}

for i in range(n):
    serial = input().strip()
    serial_len = len(serial)
    num_sum = 0
    for j in serial:
        if j.isdigit():
            num_sum += int(j)
    n_d[serial] = [serial_len, num_sum]
# print(n_d.items())
n_d = sorted(n_d.items(), key = lambda x : (x[1][0], x[1][1], x[0]))

for i in n_d:
    print(i[0])