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

출처 : www.acmicpc.net/problem/1181

1181번: 단어 정렬

우선 중복되는 것을 없애주기 위해 set을 이용하여 중복을 제거해주었다.

길이가 짧은 것을 우선으로 정렬하므로 각 단어의 길이와 단어 튜플을 sorted_a 배열에 넣었다.

파이썬 내장함수 sort()는 튜플 중 첫번째 원소를 기준으로 정렬하고 같을 경우 다음 원소를 기준으로 오름차순으로 정렬한다.

이 때문에 단어의 길이를 튜플의 첫번째 원소로 넣어주었다.

import sys
 
N = int(sys.stdin.readline().strip('\n'))
 
a = []
 
for i in range(N):
    s = sys.stdin.readline().strip('\n')
    a.append(s)
    
a = list(set(a))
 
 
sorted_a = []
 
for i in a:
    sorted_a.append((len(i),i))
 
sorted_a.sort()
 
for len_s,s in sorted_a:
    print(s)