[baekjoon] 백준 1431번(파이썬): 시리얼 번호

문제

1431번: 시리얼 번호

알고리즘

- 정렬을 통해 문제를 수행한다.

- lambda를 통해 문자의 길이, 숫자의 합, 사전 순 <= 순으로 정렬한다.

- 정렬한 문자를 출력한다.

코드

import sys


# 숫자만 더한다.
def sum_num(num):
    res = 0
    for word in num:
        if word.isdigit():
            res += int(word)
    return res


n = int(sys.stdin.readline())
m = [list(sys.stdin.readline().strip()) for _ in range(n)]

# 정렬(길이, 숫자의 합, 사전순)
m.sort(key=lambda x: (len(x), sum_num(x), x))
for i in m:
    print("".join(i))

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법