[백준] 1181번 단어 정렬

문제 출처 : https://www.acmicpc.net/problem/1181

정답 코드 💻

import sys

n = int(sys.stdin.readline())
word =[]
for _ in range(n) :
    word.append(sys.stdin.readline().strip())

set_word = set(word)
word = list(set_word)
word.sort()
word.sort(key=len)
for i in word :
    print(i)

설명

👉🏻 출력 속도를 위해서 sys.stdin.readline()을 사용하고 readline은 \n\'을 포함하는 입력이기 때문에 .strip()을 추가로 작성해야한다. 중복을 제거해야해서 set함수로 지워주고 길이로 정렬하기 위하여 sort(key = len)을 작성한다.

느낀 점 ✏️

마냥 속도를 위해서 readline()을 사용하는 것이 아니라 공백을 제거해야하고 길이로 정렬을 하려면 ()안에 조건을 넣으면 된다는 것을 배웠다.