[백준] 1431번 시리얼 번호

문제 설명

다솜이는 기타를 많이 가지고 있다. 그리고 각각의 기타는 모두 다른 시리얼 번호를 가지고 있다. 다솜이는 기타를 빨리 찾아서 빨리 사람들에게 연주해주기 위해서 기타를 시리얼 번호 순서대로 정렬하고자 한다.

모든 시리얼 번호는 알파벳 대문자 (A-Z)와 숫자 (0-9)로 이루어져 있다.

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

1.A와 B의 길이가 다르면, 짧은 것이 먼저 온다.

2.만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)

3.만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.

시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.

제한 사항

-첫째 줄에 기타의 개수 N이 주어진다. N은 1,000보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어져 있다. 시리얼 번호는 중복되지 않는다.

​-첫째 줄부터 차례대로 N개의 줄에 한줄에 하나씩 시리얼 번호를 정렬한 결과를 출력한다.

​

입출력 예

input	output
5 ABCD 145C A A910 Z321	A ABCD Z321 145C A910

​

접근법

re 라이브러리를 이용하면 굉장히 쉽게 풀립니다. 입력 받을 때, 입력 받은 문자열 내에 숫자가 있는지 없는지 re.findall()함수를 이용하여 찾아줍니다. 만약에 있을 경우에 그 숫자들이 문자 리스트로 반환되어, 각 숫자형태의 문자를 정수형으로 바꾸어준 뒤, sum을 이용하여 합해줍니다. 만약 숫자를 포함하지 않을 경우에는 0이됩니다. 이 결과를 입력 리스트에 (입력받은 시리얼 번호, 숫자의 합) 형태로 넣어줍니다. 이후 정렬을 해주는데, 첫 번째 기준으로는 시리얼 번호의 길이, 두 번째는 숫자, 세 번째는 입력받은 시리얼 번호로 설정해 주시면 됩니다.

​

나의 코드

import sys
import re

number = int(sys.stdin.readline())
serial_numbers = []
for _ in range(number):
    input_serial = sys.stdin.readline().replace('\n','')
    num_sum = sum(list(map(int, re.findall('\d',input_serial))))
    serial_numbers.append((input_serial,num_sum))

serial_numbers.sort(key=lambda x:(len(x[0]),x[1],x[0]))
for serial in serial_numbers:
    sys.stdout.write(serial[0]+"\n")