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

1181번: 단어 정렬

import sys
n = int(sys.stdin.readline().rstrip())
arr = [sys.stdin.readline().strip() for _ in range(n)]
new_arr = [(i, len(i)) for i in set(arr)]
ans_list = sorted(new_arr, key=lambda x : (x[1], x[0]))

for i in ans_list :
    print(i[0])

 이번 문제는 크게 어렵지 않다.

1. 여러 번 입력된 경우 하나만 출력하라 했으니, 배열을 입력받아서 set으로 중복을 없애준다. 
2. 다음 list comprehension을 통해 길이 feature를 하나 추가해준다
3. 마지막으로 길이에 대해서 sort먼저, 그 후 사전 순으로 sorting 해주면 문제 마무리