[백준/Python] 1431번 - 시리얼 번호

문제

백준 1431번 - 시리얼 번호

해결 과정

• sort 사용하기
• 모든 자리수의 합을 구하는 함수를 구현
  • isdigit() 사용

시행착오

풀이

import sys
n = int(sys.stdin.readline())

array = []

# 모든 자리수의 합을 구하는 함수
def sum_num(str):
  result = 0
  for i in str:
    if i.isdigit():
      result += int(i)
  return result
  
for _ in range(n):
  array.append(sys.stdin.readline().strip())

array.sort(key = lambda x:(len(x),sum_num(x),x))

for i in array:
  print(i)