백준 영단어 암기는 외로워[20920번]

문제

백준 영단어 암기는 외로워

화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다.

그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다.

화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.

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

M 보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가 M 이상인 단어들만 외운다고 한다.

화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.

입력

첫째 줄에는 영어 지문에 나오는 단어의 개수 N과 외울 단어의 길이 기준이 되는 M이 공백으로 구분되어 주어진다. (1 ≤ N ≤ 100,000, 1 ≤ M ≤ 10)

둘째 줄부터 N + 1번째 줄까지 외울 단어를 입력받는다.

이때의 입력은 알파벳 소문자만 주어지며 단어의 길이는 10을 넘지 않는다.

단어장에 단어가 반드시 1개 이상 존재하는 입력만 주어진다.

출력

화은이의 단어장에 들어 있는 단어를 단어장의 앞에 위치한 단어부터 한 줄에 한 단어씩 순서대로 출력한다.

예제 입출력

입력 1

7 4
apple
ant
sand
apple
append
sand
sand

출력 1

sand
apple
append

입력 2

12 5
appearance
append
attendance
swim
swift
swift
swift
mouse
wallet
mouse
ice
age

출력 2

swift
mouse
appearance
attendance
append
wallet

풀이

# 0. m개 이상인 단어만 가져간다
# 1. 딕셔너리로 영단어:카운트갯수로한다
# 2. 만약 카운트 갯수가 같으면 => 단어의길이를 검사해서 긴거를 앞에배치한다
# 3. 카운트갯수도 같고 단어길이도 같으면=> 알파벳 사전순으로 배치한다

import sys
from collections import defaultdict
num, length= map(int,sys.stdin.readline().rstrip().split(" "))
alpha_count = defaultdict(int)

# 길이가 m개 이상인 단어만 저장
for _ in range(num):
    alpha=sys.stdin.readline().rstrip()
    if len(alpha)>=length:
        alpha_count[alpha]+=1

      # 1, 2, 3에 해당하는 조건을 적용
result = sorted(
    alpha_count.items(),
    key=lambda item: (-item[1], -len(item[0]), item[0])
)

result=[alpha for alpha,_ in result]

for alpha in result:
    print(alpha)

설명

주어진 조건에 따라 iterable한 자료구조를 정렬하면 되는 문제이다.

필자는 defaultdict를 활용하였다.

defaultdict는 일반 dict과는 다르게 들어갈 value값 유형을 정해주면 기본값으로 0, “” 등이 들어간다.