[백준 파이썬] #1431: 시리얼 번호

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

1431번: 시리얼 번호

 

[문제 정리]

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

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

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
2. 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
3. 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.

[문제 접근]

정렬 문제이므로 sort()함수을 사용한다. 기준이 여러 개이므로 key를 사용한다.

 

[정답]

 

1. 기타의 개수 N을 입력받는다. 시리얼 번호를 저장할 리스트 num을 정의한다.

2. for문을 돌면서 num에 시리얼 번호를 입력받아 추가한다.

3. 일단 기준 3번을 충족하기 위해 num을 sort()함수를 사용해 정렬한다. 그런 다음 기준 1,2번을 충족하기 위해 key를 사용한다.

num.sort( key=lambda x: ( len(x) ,sum(int(a) for a in x if a.isnumeric() ))) : 1번 기준인 길이 len(x)와 2번 기준인 숫자인 모든 자리수의 합 sum(int(a) for a in x if a.isnumeric())

4. for문을 사용해 num을 하나씩 출력한다.