시리얼 번호 백준 1431번 정렬

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

문제

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

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

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

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

순서대로 하면 됩니다. 1번에서는 길이를 기준으로 정렬합니다. 그러면 길이를 기준으로 하는 값을 하나 저장합시다.

그 다음에는 자리수의 합이 먼저 오는 것으로 합니다. 마지막으로 사전순으로 비교를 한다고 합니다.

num = int(input())
dit_list = []
for i in range(num):
    temp = str(input())
    lenght = len(temp)
    count = 0
    for j in range(lenght):
        try:
            count += int(temp[j])
        except ValueError:
            pass
    dit_list.append([lenght,count,temp])
    
dit_list.sort(key=lambda x: (x[0],x[1],x[2]))
for k in range(num):
    print(dit_list[k][2])

lenght는 길이를 담는 변수입니다. 그리고 count에서 try except를 사용하는데 정규식 re 사용할 것 없이 간단하게 문자열에서 숫자만 가져올 수 있습니다. int자료형변환을 할때 가능하다면 숫자이고 아니라면 문자열이겠지요

이때 try를 해서 시도를 하고 만약 ValueError이 일어나면 pass 그냥 넘어가라고 제어문을 달았습니다. 그러면 에러없이 코드가 넘어갈 수 있습니다. 그렇게 count 숫자의 크기도 저장을 했습니다. 이렇게 3가지를 key에다가 정렬을 하라고 주면 끝납니다.