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

문제

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

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

단, 중복된 단어는 하나만 남기고 제거해야 한다.

 

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000)

둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다.

주어지는 문자열의 길이는 50을 넘지 않는다.

 

출력

조건에 따라 정렬하여 단어들을 출력한다.

 

 

문제 풀이

본 문제는 간단히 정렬로 해결할 수 있는 문제이다.

문제에서 주의해서 볼 것은 두 가지인데,

 

1. 중복된 단어는 하나만 남기고 제거한다.

2. 길이로 정렬하되, 길이가 같은 경우 사전 순으로 정렬한다.

 

이는 파이썬 정렬 함수들에 대한 이해가 부족하면 조금 어려울 수 있다.

 

기본적으로 파이썬의 sorted, sort 함수는 문자열에 대해 사전순으로 정렬한다.

따라서, 길이와 사전순 정렬로 두 가지를 고려한 Key를 정의해주면 쉽게 해결할 수 있다.

 

이어서 중복된 단어를 제거해야 하는데, 제거하는 것은 간단하다.

 

모든 문자를 파이썬 집합으로 받아내는 것이다.

 

구현한 코드를 보면 아래와 같다.

 

정답코드

n = int(input())

data = set()
for _ in range(n):
    data.add(input())
sorted_data = sorted(data, key=lambda x: (len(x), x))

for word in sorted_data:
    print(word)