[알고리즘][Python] 백준(BOJ) 1431 시리얼 번호_파이썬

 

1431 시리얼 번호    www.acmicpc.net/problem/1431

1431번: 시리얼 번호

 

 

문제 풀기 전 공부할 것 : 정렬

 

 

 

 

 

 

 

풀이

<내용>

• n과 시리얼 번호를 입력받는다.
• 모든 자릿수의 합(숫자만)을 구하는 sum_num 함수를 정의한다.
• 시리얼 번호를 lambda 함수를 통해 길이, 모든 자릿수의 합, 사전 순으로 정렬한다.
• 시리얼 번호를 하나씩 출력한다.

 

 

<코드>

import sys
input = sys.stdin.readline

def sum_num(s):
    res = 0
    for i in s:
        if i.isdigit():
            res += int(i)
    return res

n = int(input())
serial = [input().rstrip() for _ in range(n)]
serial.sort(key= lambda x:(len(x), sum_num(x), x))
for s in serial:
    print(s)