[Python]백준_1431 : 시리얼 번호

2022년 12월 14일에 작성됨

 

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

1431번: 시리얼 번호

 

 

문제 분석

n개의 시리얼 번호 문자열을 입력받고, 기준에 따라 정렬하여 출력한다.

1. 길이가 짧은 순서
2. 모든 자리수의 숫자만 더한 값이 작은 순서
3. 사전 순서

 

 

소스 코드 (⭕)

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)

 

코드 분석

1. sum_num 사용자 정의 함수를 선언하고, 매개변수로 받을 arr의 각 요소 i가 모두 숫자로만 구성되어 있으면 result에 i의 자릿수 합을 구하고 반환해준다.

2. n을 입력받고, n번 만큼 arr 리스트에 문자열을 입력한다.

3. key 인자에 x를 매개변수로 하는 lambda 함수를 할당하여 각각 길이, 자릿수의 합, 사전 순으로 arr을 오름차순 정렬을 해준다.

 

 

 

✍️isdigit() 함수

'1234'처럼 문자열의 각 문자들이 모두 숫자로 되어있을 때 true를 리턴하고, 숫자가 아닌 알파벳과 같은 문자가 끼어있으면 false를 리턴한다.

✍️lambda() 함수

<기본 형태>
lambda 인자 : 표현식
ex) lambda x : x+1

key 매개변수를 가지는 sort() 함수는 key 값을 기준으로 정렬되고 기본값은 오름차순이고, lambda식을 사용할 수 있다.

 

 

end

lambda 함수 은근 어렵고 헷갈린다..ㅠ_ㅠ 그래도 검색해볼 때마다 점점 이해가 되고 있다.!!