[백준/BOJ] 1431번: 시리얼 번호 (파이썬3/Python3)

문제

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

1431번: 시리얼 번호

---

생각

우선 정렬의 기준이 길이가 먼저, 그 다음은 자리수의 합(숫자만), 그 다음은 사전 순이다.

길이나 사전 순은 lambda의 기준에 그냥 넣으면 되지만 자리수는 다르다!

우움 자리수를 미리 더해서 리스트에 추가하거나 함수를 만드는 것이 나을 것 같다. 

---

코드

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

N = int(input())
R = []
for i in range(N):
  R.append(input())

R.sort(key=lambda x:(len(x), sum(x), x))
for i in R:
  print(i)