Hello Computer Vision

백준 알고리즘 1181번 [단어 정렬] 파이썬 본문

백준알고리즘

백준 알고리즘 1181번 [단어 정렬] 파이썬

지웅쓰 2023. 4. 23. 14:59

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

단어를 정렬하는 것인데 핵심은 단어 개수를 먼저 봐야한다는 것이다. 그리고 나서 단어 개수가 같다면 알파벳순으로 정렬하라는 것인데 상위 정렬이 개수이고, 하위 정렬이 알파벳인걸 알 수 있다. 그렇다면 코드에서는 먼저 하위 정렬 개념인 알파벳 정렬을 해주고 상위 정렬인 개수로 정렬을 해주면 된다.

n = int(input())
word_list = []
for _ in range(n):
    word_list.append(input())
word_list = set(word_list)
word_list = list(word_list)

word_list.sort()
word_list.sort(key = len)

for i in word_list:
    print(i)