[BOJ] 1181. 단어 정렬

문제

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

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

• 단 중복 단어는 제거

 

---

 

주요 아이디어

• cmp_to_key 라이브러리에 사용자 정렬 조건을 전달해 풀었다

 

---

 

코드 구현 (Python 3)

from functools import cmp_to_key

def compare(w1, w2):
    if len(w1) < len(w2):
        return -1
    elif len(w1) > len(w2):
        return 1
    else:
        if w1 < w2:
            return -1
        else:
            return 1

n = int(input())
word_list = [input() for _ in range(n)]

# 중복 제거
word_list = sorted(list(set(word_list)), key=cmp_to_key(compare))

for word in word_list:
    print(word)

 

---

 

제출 결과