백준 1431 - 시리얼 번호 (Python)

문제

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

1431번: 시리얼 번호

풀이

a = []
for _ in range(int(input())):
    s = input()
    a.append([len(s),sum([int(i) for i in s if i.isdigit()]),s])

a.sort(key=lambda x: (x[0],x[1],x[2]))

for i in a:
    print(i[2])

코드는 많이 줄여서 썼지만 알고리즘은 다음과 같다.

[문자열의 길이, 숫자인 것들의 합, 문자열 전체] 이렇게 이루어진 리스트를 원소로 가지는 이차원 리스트를 구성한 뒤,

0번째, 1번째, 2번째 인덱스의 값 순서로 우선순위를 두어 정렬한다.

이후 문자열에 해당하는 2번째 값을 출력한다.