[백준 1181] 단어 정렬 - Python

문제

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

단어 정렬

 

풀이

n = int(input())
words = []
for _ in range(n):
    words.append(str(input()))
words = set(words)
answer = sorted(words, key = lambda x: (len(x), x))
for i in answer:
    print(i)

for문을 통해 주어진 문자열을 str 형식으로 받는다. 이후 set() 함수를 통해 중복된 단어를 제거하고, lambda 함수를 통해 정렬 시 첫 번째 조건을 단어의 길이로 오름차순, 이후 사전 순 오름차순으로 설정하였다.