코딩테스트 대비/BOJ
[Baekjoon/Python] 1181번: 단어 정렬 - 효과는 굉장했다!
bluetag_boy
2021. 10. 10. 21:34
반응형
1181번: 단어 정렬
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
www.acmicpc.net
알고리즘 분류
- 정렬
SOLUTION
import sys
words_list = set() #중복된 단어는 삭제하기 위해 set으로 할당
for _ in range(int(sys.stdin.readline())):
words_list.add(sys.stdin.readline().rstrip()) # set은 list와 다르게 append() 함수 대신 add() 함수 사용, rstrip을 통해 문자열 뒤 \n 붙는 것 방지
words_list = list(words_list) # sort 함수 사용을 위해 리스트 화
# 알파벳 순서로 정렬 -> 알파벳 순서로 정렬된 리스트를 다시 짧은 단어 순으로 정렬
# Why? 길이로만 비교하면 단어의 알파벳 순서로 정렬되지 않으니까
words_list.sort()
words_list.sort(key = lambda word: len(word))
print("\n".join(words_list)) # 줄 바꿈 \n 과 join 함수를 통해 각 줄당 단어 하나씩 출력