[Daily 코테] 백준 1181번: 단어 정렬

 

n = int(input())
words = set([input() for _ in range(n)])

words = sorted(list(words))
words = sorted(words, key=lambda x: len(x))
for i in range(len(words)):
    print(words[i])

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

1181번: 단어 정렬

First Try

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

입력받을 단어의 개수를 변수 n에 저장하고, words라는 리스트에 n번 반복하여 단어를 입력받는다.

문제에서 제시한 정렬 기준은 첫번째 우선 순위가 문자의 길이였고, 두번째 우선 순위는 문자의 알파벳 순서였다.

즉, 문자들을 문자의 length 순서대로 모두 정렬한 후, 정렬된 리스트에서 길이가 같은 문자들을 알파벳 순서대로 또 정렬해주면 된다.

이때 위 코드에서 알파벳 순서대로 먼저 정렬한 이유는 python 내장 함수 sorted()의 특성 때문이다. 

세번째 줄과 네번째 줄을 바꾸면 에러가 발생한다.

sorted()는 Tim sort라는 정렬 방식을 사용한다. 필자도 문제를 풀면서 확실한 원리를 잘 알지 못하여 Tim sort에 대해 찾아봤다.

https://www.youtube.com/watch?v=2pjUsuHTqHc 

 

⚪이슈 원인

set은 인덱스가 없다. 마지막 for문에서 set의 인덱스를 이용해서 단어를 출력하려다 에러가 발생했다.

Second Try🎉

n = int(input())
words = set([input() for _ in range(n)])

words = sorted(list(words))
words = sorted(words, key=lambda x: len(x))
for i in range(len(words)):
    print(words[i])