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

문제

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

1431번: 시리얼 번호

---

시리얼번호 A가 시리얼번호 B의 앞에 오는 경우는 다음과 같다.

A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.
시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.

문제 中

 

문자열이 1~50개 주어지고, 이를 주어진 조건에 따라 정렬하는 문제이다. 

 

정렬 조건

1. 길이가 짧은 문자열이 앞에 오게 한다. 

2. 1번 조건이 같다면 문자열에 있는 숫자의 합이 적은 것이 앞에 오게 한다.

3. 2번이 같다면 사전순으로 정렬한다. 

 

---

 

n = int(input())
data = []
datasum = dict()
for i in range(n):
    a = input()
    temp = 0
    data.append(a)
    for c in a:
        if c in "123456789":
            temp += int(c)
    datasum[a] = temp

data.sort(key = lambda x : (len(x), datasum[x], x))
for i in data:
    print(i)

 

datasum리스트에 각 단어의 숫자합을 넣고, 

위 3가지 정렬 조건을 sort함수의 key인자에 람다함수를 넣어 정렬하였다.

 

data.sort(key = lambda x : (len(x), datasum[x], x))

 

len(x)는 x의 길이, datasum[x]는 x번째의 단어의 숫자합, x는 사전순으로 정렬하라는 의미이며, 이를 나열하는 것으로 문제를 해결할 수 있다.