출처
https://www.acmicpc.net/problem/1431
문제
다솜이는 기타를 많이 가지고 있다. 그리고 각각의 기타는 모두 다른 시리얼 번호를 가지고 있다. 다솜이는 기타를 빨리 찾아서 빨리 사람들에게 연주해주기 위해서 기타를 시리얼 번호 순서대로 정렬하고자 한다.
모든 시리얼 번호는 알파벳 대문자 (A-Z)와 숫자 (0-9)로 이루어져 있다.
시리얼번호 A가 시리얼번호 B의 앞에 오는 경우는 다음과 같다.
- A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
- 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
- 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.
시리얼이 주어졌을 때, 정렬해서 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 기타의 개수 N이 주어진다. N은 50보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어져 있다. 시리얼 번호는 중복되지 않는다.
출력
첫째 줄부터 차례대로 N개의 줄에 한줄에 하나씩 시리얼 번호를 정렬한 결과를 출력한다.
예제 입출력
알고리즘 분류
- 정렬
➡️문제 분석
n개의 시리얼 번호 문자열을 입력받고, 기준에 따라 정렬하여 출력한다.
- 길이가 짧은 순서
- 모든 자리수의 숫자만 더한 값이 작은 순서
- 사전 순서
➡️코드(⭕)
import sys
input = sys.stdin.readline
def sum_num(x):
result = 0
for i in x:
if i.isdigit() == True: # 문자열 i가 모두 숫자로만 구성되어 있으면
result += int(i)
return result
arr = []
n = int(input())
for _ in range(n):
arr.append(str(input().rstrip()))
# lambda 함수로 길이, 모든 자릿수의 합, 사전 순으로 오름차순 정렬
arr.sort(key=lambda x: (len(x), sum_num(x), x))
# arr = sorted(arr, key=lambda x: (len(x), sum_num(x), x))도 같은 뜻
for i in arr:
print(i)➡️코드 분석
-
sum_num 사용자 정의 함수를 선언하고, 매개변수로 받을 arr의 각 요소 i가 모두 숫자로만 구성되어 있으면 result에 i의 자릿수 합을 구하고 반환해준다.
-
n을 입력받고, n번 만큼 arr 리스트에 문자열을 입력한다.
-
key 인자에 x를 매개변수로 하는 lambda 함수를 할당하여 각각 길이, 자릿수의 합, 사전 순으로 arr을 오름차순 정렬을 해준다.
✍️isdigit() 함수
'1234'처럼 문자열의 각 문자들이 모두 숫자로 되어있을 때 true를 리턴하고, 숫자가 아닌 알파벳과 같은 문자가 끼어있으면 false를 리턴한다.
✍️lambda() 함수
<기본 형태>
lambda 인자 : 표현식
ex) lambda x : x+1
key 매개변수를 가지는 sort() 함수는 key 값을 기준으로 정렬되고 기본값은 오름차순이고, lambda식을 사용할 수 있다.
➡️end
lambda 함수 은근 어렵고 헷갈린다..ㅠ_ㅠ 그래도 검색해볼 때마다 점점 이해가 되고 있다.!!
