백준 1181번 : 단어 정렬(python)

문제링크
https://www.acmicpc.net/problem/1181

코드

import sys
N=int(input())
arr = []
word = [sys.stdin.readline().strip() for i in range(N)]
for i in word:
    arr.append((len(i),i))
a1 = sorted(list(set(arr)))
for i in a1:
    print(i[1])

시간을 줄이기 위해서 입력방식을 input에서 sys를 이용한 방식으로 바꿨습니다.
단어 길이와 단어를 묶어서 arr 리스트에 넣었습니다. 중복값을 없애주기 위해서 set()을 사용한 후 list로 만들었습니다. 그리고 sorted()를 이용해서 정렬된 a1 리스트를 만들어 출력했습니다.

sorted()와 sort()차이

sorted를 사용하면 정렬된 새로운 리스트를 만드는 것이고 sort는 기존의 리스트를 정렬하는 것입니다.