[Python] 백준/BOJ 20920번: 영단어 암기는 괴로워 (Silver 3)

💻 Problem

문제 보러 가기

 

화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다. 그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다. 화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.

1. 자주 나오는 단어일수록 앞에 배치한다.
2. 해당 단어의 길이가 길수록 앞에 배치한다.
3. 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치한다

 M보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가 M이상인 단어들만 외운다고 한다. 화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.

 

💡 Approach

먼저 딕셔너리에 단어와 그 단어의 빈도수를 저장했다.

그 후 sorted 함수에서 lambda를 활용해 정렬했다.

• −x[1]: 빈도 내림차순
• −len(x[0]): 단어 길이 내림차순
• x[0]: 알파벳 오름차순

 

✏️ Solution

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
vocabulary = {}

for _ in range(n):
    word = input().rstrip()
    
    if len(word) >= m:
        vocabulary[word] = vocabulary.get(word, 0) + 1

# 1. 빈도 내림차순
# 2. 단어의 길이 내림차순
# 3. 알파벳 오름차순
vocabulary = sorted(vocabulary.items(), key=lambda x: (-x[1], -len(x[0]), x[0]))

for word in vocabulary:
    print(word[0])

시간복잡도: O(n log n)