[Python] 시리얼 번호 - 백준 (정렬)

해당 글은 백준 1431번 문제 '시리얼 번호'를 다룬다.

문제

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

1431번: 시리얼 번호

 

설명

해당 문제는 각 조건의 기준에 맞게 정렬하는 문제이다.

이는 lambda 식을 이용하면 한 번에 여러개의 조건을 지정해 줄 수 있다.

lambda x:(조건1, 조건2, 조건3, ...) => 조건1이 같다면 다음 조건2를 비교하는 형태로 진행된다.

 

코드

n = int(input())
serial = []
for i in range(n):
    serial.append(input())

def sum(x):
    sum = 0
    for i in x:
        if i.isdigit():
            sum += int(i)
    return sum

serial.sort(key=lambda x:(len(x), sum(x), x))
for i in range(n):
    print(serial[i])