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

< 기존 풀이 >

함수 구현을 직접 해서 해결 하려고 하였고 당연히 시간 초과가 발생했다.

 

< 해결 방법 >

sort 중 lambda 를 이용한 해결을 공부했었고 이를 응용해서 풀었다. lambda 사용 시 key = ~~ 라고 적는 것을 몰라서 답지를 참조했다. 유용한 방법인 것 같다. 

 

import sys
 
a = sys.stdin.readline()
 
b = [ ]
 
 
for i in range(0,int(a)):
    b.append(sys.stdin.readline().rstrip())
 
b=sorted(set(b))
b=sorted(b,key=lambda x : len(x))
 
for i in b:
    print(i)