본문 바로가기

파이썬

[백준] 1181번 단어정렬 (파이썬)

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

 

1181번: 단어 정렬

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

www.acmicpc.net

 

<풀이>

N = int(input())
words = []
for i in range(N):
    words.append(input())

words = list(set(words))
words.sort()
words.sort(key=lambda x:len(x))
for word in words:
    print(word)

 

<풀이방법>

1. 중복단어 제거해야 하므로 set을 사용하였고 

2. 길이가 같을 경우 사전 순으로 sort를 사용하였음

3. 길이순서대로 정렬해야하므로 lambda x: len(x)로 하여서 길이가 짧은 순으로 정렬하였다.