[백준] 1181번 단어 정렬 - 파이썬

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

1181번: 단어 정렬

 

풀이

정렬을 두 번 해주었습니다. 사전순으로 오름차순으로 정렬을 해준 뒤에 길이를 기준으로 정렬을 추가로 진행해주었습니다.

 

1. 오름차순으로 정렬(문제의 테스트 케이스 기준)

but
cannot
hesitate
i
im
it
more
no
wait
wont
yours

2. 오름차순에 이어서 길이 기준으로 정렬

i
im
it
no
but
more
wait
wont
yours
cannot
hesitate

 

# 1181번 단어 정렬
import sys
n = int(input())
s = set()
for _ in range(n):
    s.add(sys.stdin.readline())
s = list(s) 
s.sort()	# 사전순으로 정렬
s.sort(key=len)	# 길이기준으로 정렬
print(''.join(s))

출력결과