[백준, BOJ 1181] 단어 정렬 (python)

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

1181번: 단어 정렬

---

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

출력

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

728x90

예제 입력 1

13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours

예제 출력 1

i
im
it
no
but
more
wait
wont
yours
cannot
hesitate

---

# 문제 핵심
# 1. 중복된 단어는 한 번만 출력
# 2. 문자열 길이로 오름차순 정렬
# 3. 문자열 길이가 같을 때는 사전 순으로 정렬

N = int(input())
str = []

# 단어들 입력
for i in range(N):
    str.append(input())
    
# set의 중복이 없다는 특징을 이용해서 단어들의 중복 제거
set_str = list(set(str))

# (단어 길이, 단어) 형태로 새로운 리스트에 저장.
# [(5, adfdf), (2, ba), (2, ab)] 형식으로
len_info_str = []
for i in set_str:
    len_info_str.append((len(i), i))
    
# 이를 sorted 함수로 정렬하면
# 우선 len을 기준으로 정렬하고,
# 다음으로 사전 순으로 정렬된다.
sorted_str = sorted(len_info_str)

# 현재 sorted_str은
# [(2, ab), (2, ba), (5, adfdf)]와 같이 정렬되어 있으므로
# 뒤의 단어만 따로 저장해준다.
result = []
for lgt, word in sorted_str:
    result.append(word)

# 출력
for word in result:    
    print(word)