[백준 1181] 단어 정렬 (python3)

 

ASAC 과정에서 정렬에 대해서 공부하면서 함께 연습문제로 풀어보기 위해 간단한 단어 정렬 문제를 풀어보았다.

# 단어의 개수를 받을 수 있는 N 입력 받기
N = int(input())

# 리스트 저장
words = []

for i in range(1, N+1, 1):
    words.append(str(input()))

#중복제거 후 리스트 변환
answer = list(set(words))

# 길이수, 알파벳 오름차순 정렬
answer.sort(key=lambda x : (len(x), x))

#출력
for i in answer:
    print(i)

 

 

내 코드는 이렇게 작성하였는데, 시간이 조금 많이 걸린 느낌이었다. 아마도 input을 중복으로 사용하면서 파이썬에서 시간이 더 많이 잡았던 것으로 보인다. input 함수가 아닌 sys.stdin.readline을 통해 입력 받아보면 더욱 시간이 줄어들 것 같은데, 이 부분은 시간 날 떄 다시 작성해보면서 여기에 추가해봐야겠다.