백준 1181 단어 정렬 sorted, lambda (파이썬)

아이디어

1. 그냥 정렬할 때 lambda 써보라고 만든 문제같다.
2. 정렬할 때, key를 lambda로 주고, 선순위, 후순위를 만들면 될 것 같다.

---

풀이 및 해설(주석)

import sys

N = int(sys.stdin.readline())
arr = set()
for i in range(N):                  # 중복된 원소가 들어가면 안되니까 set으로 넣어준다.
    arr.add(sys.stdin.readline().rstrip())
list(arr)
result = sorted(arr, key=lambda x:(len(x), x))   # 정렬에 lambda사용, 1순위는 길이, 2순위는 알파벳순...

for r in range(len(result)):
    sys.stdout.write(result[r])     # 처음 sys.stdout.write를 써봣다.
    sys.stdout.write('\n')          # 이거는 기본적으로 뒤에 개행문자가 안들어가 있는 것 같다.