백준 1431번 시리얼 번호 (Python)

문제

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

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

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

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

문제 해결 접근

• 파이썬 리스트 정렬 함수와 기본 정렬 알고리즘을 모두 활용해볼 수 있는 문제
• 먼저, N<=50 이므로 O(n^2)까지 충분히 가능함으로 선택,삽입,버블 정렬도 사용할 수 있다.
• 조건3은 1,2번으로 해결하지 못했을 때 그대로 놓여지고 있으면 되기 때문에, 오히려 조건3을 가장 먼저 적용해준 뒤, 조건 1, 조건 2 순서대로 풀 수 있게 하기
• 조건 1,3은 파이썬의 sorted, lambda 함수 사용하기
  사전 순으로 정렬하기 : array = sorted(array)
  길이 순으로 정렬하기 : array = sorted(array, key=lambda x: len(x))
• 조건2는 어떤 것으로 풀어도 되나, 나는 🫧버블🫧 정렬이 좋아서 🫧버블🫧 정렬 사용 (이름이 귀여우니까)
  re를 사용해서 숫자를 뽑아내고 합을 비교할 수 있다.

정답 코드

import sys
import re

N = int(sys.stdin.readline())  # N<=50, O(n^3)까지 가능 
array = []

for i in range(N):
  array.append(sys.stdin.readline().strip())

## 조건 3 : 서로 길이도 같고, 모든 자리수의 합이 같을 때 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.
array = sorted(array)

## 조건 1 : 길이가 짧은 거 순서로 정렬
array = sorted(array, key=lambda x: len(x))

## 조건 2 : 길이가 같은 경우 자리 수의 합을 비교해서 작은 합을 가지는 것이 먼저 오도록
## 버블 정렬 
for i in range(N-1):
  for j in range(N-i-1):
    if len(array[j]) == len(array[j+1]):
      hap_j = sum([int(j) for j in array[j] if re.search('[0-9]', j)])
      hap_jj = sum([int(j) for j in array[j+1] if re.search('[0-9]', j)])
      if hap_j > hap_jj:
        array[j], array[j+1] = array[j+1], array[j]	
      
for i in array:
  print(i)

복습할 알고리즘

정렬 알고리즘 다시 짚고 넘어가자!

1) 선택 정렬 : 쭈욱 훑으면서서 제일 작은 애 찾기! 그렇게 찾은 가장 작은 애를 앞으로 보내기!

for i in ragne(len(array)):
	min_index = i
    for j in range(i+1, len(array)):
    	if array[min_index] > array[j] :
        	min_index = j
    array[i], array[min_index] = array[min_index], array[i]
    

2) 삽입 정렬 : 뒤에서부터 훑어가면서 나보다 작은 게 있으면 바꾸기 (데이터가 거의 정렬되어 있을 때 효과적)

for i in range(1, len(array)):
	for j in range(i, 0, -1):
    	if array[j] > array[j-1]: ## 나보다 큰애들이 앞에 있으면 재끼고 앞으로 가기
        	array[j], array[j-1] = array[j-1], array[j]
        else: ## 더이상 옮기지 않으면 break
        	break

3) 버블 정렬 : 인접한 두 수를 비교하며 정렬해나가기!

for i in range(n - 1):
	for j in range(n - i - 1):
		if array[j] > array[j + 1]:
			array[j], array[j + 1] = array[j + 1], array[j]