<정렬> 백준 1181번. 단어 정렬

https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

 

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로

조건대로 정렬하면 된다.

 

그렇다면, 사전순으로 정렬해주고 길이순으로 정렬해주면 될 것이다.

import sys
input = sys.stdin.readline

n = int(input())
lst = []
for i in range(n):
  lst.append(input().rstrip("\n"))
lst = list(set(lst))
lst.sort()
lst.sort(key=len)
print(*lst, sep = "\n")

정답이긴 하지만, 정렬을 두 번해서 메모리와 시간이 괜찮을지 의문이다.