문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
출력
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.
코드
import sys
n = int(sys.stdin.readline())
arr = []
for i in range(n):
arr.append(sys.stdin.readline().strip())
#중복 제거 후 sort
set_arr = list(set(arr))
set_arr.sort()
set_arr.sort(key = len)
for i in set_arr:
print(i)
풀이
문제 정렬 조건이 다음과 같다.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
그렇다면, 정렬 순서가 하위 조건인 사전 순으로 정렬을 하고 상위 조건인 길이가 짧은 것부터 정렬을 해야한다.
하위 조건 먼저!!
* 문자열 길이로 sort 하는 경우
arr.sort(key = len)
'알고리즘 문제 풀이 > 백준' 카테고리의 다른 글
[백준] 11651 좌표 정렬하기2 python (0) | 2022.02.15 |
---|---|
[백준] 2309 일곱 난쟁이 python (0) | 2022.02.12 |
[백준 알고리즘] 10989 수 정렬하기3 python (0) | 2022.02.12 |
[백준] 16953 A -> B python (0) | 2022.02.10 |
[백준] 4769 캠핑 python (0) | 2022.02.10 |