본문 바로가기

백준

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

반응형

1. 문제

2. 코드

n = int(input())
l = []
for i in range(n):
    s = input()
    cnt = 0
    for j in range(len(s)):
        if 48<= ord(s[j]) <=57:
            cnt += int(s[j])
    l.append([s, cnt])

l = sorted(l, key = lambda x: (len(x[0]), x[1], x[0]))

for i in range(n):
    print(l[i][0])

 

3. 주요 알고리즘

lambda를 써서 sort할 때, 여러 조건들을 한번에 부여하면 된다.

 

728x90
반응형