알고리즘

[백준] 1431 - 시리얼 번호

j9972 2023. 4. 7. 16:38
728x90

1. 문제

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

 

1431번: 시리얼 번호

첫째 줄에 기타의 개수 N이 주어진다. N은 50보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어

www.acmicpc.net

 

2. 접근

1. 데이터를 받을 때 각 문자열 값 안에서 숫자의 합을 구해주는 함수를 구해준다.

2. 길이 -> 숫자의 합 -> 사전순 의 순서로 정렬을 해주기 위해 lambda 를 적용한다

 

3. 풀이

import sys
input = sys.stdin.readline

serial_num = []
for i in range(int(input())):
    serial_num.append(input().rstrip())

def count(str):
    tot = 0
    for i in str:
        if i.isdigit():
            tot += int(i)
    return tot

serial_num.sort(key=lambda x:(len(x),count(x),x))


for s in serial_num:
    print(s)