1181번 단어 정렬
이 또한 간단한 문제지만 새로운 함수 사용 기념으로 기록한다.
이런식으로 글을 작성하면 앞으로 수많은 쉬운 문제들이 올라오는게 아닌가 싶다..

코드

import sys

N = int(sys.stdin.readline())
words = []
cnt = 1

for i in range(N):
    words.append(sys.stdin.readline().strip())
set_list = set(words)
words = list(set_list)

words.sort()
words.sort(key=len)

N = len(words)
for j in range(N):
    print(words[j])

sys.stdin.readline()을 사용하면 문자열을 입력받을 때 개행 문자가 입력되는 것을 확인하였다. 이는 strip()을 사용하면 해결된다.

set함수를 사용한 이유는 리스트의 중복되는 숫자를 없애기 위함이다.
파이썬 내장 함수 sort에 길이로 정렬해주는 기능이 있다는 것을 처음 알았다.
이중 포문으로 출력하다가 시간 초과가 나와서 알아보니 정말 편리한 기능이 있었다.. 역시 파이썬은 놀라워

댓글남기기