[백준 - python] 1181번: 단어 정렬

Interrobang 2022. 10. 2. 13:24

<문제 링크>

1181번: 단어 정렬

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

www.acmicpc.net

<문제 풀이>

N=int(input())
word_list=[]

for i in range(0,N):
    word_list.append(input())
    
word_list=list(set(word_list))
word_list.sort()
word_list.sort(key=len)

for word in word_list:
    print(i)

*key point: sort 함수의 key 옵션을 이용하여 길이에 따른 정렬을 수행할 수 있다. set 함수를 이용해서는 중복을 제거할 수 있다.