백준20920 문제 보러가기

문제

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

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

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

입력

첫째 줄에는 영어 지문에 나오는 단어의 개수 $N$과 외울 단어의 길이 기준이 되는
$M$이 공백으로 구분되어 주어진다. ($1 \leq N \leq 100\,000$, $1 \leq M \leq 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

문제 해결 과정

내가 생각했을 때 정렬을 할 때 기준이 되는 것들은 너무나 쉽게 구할 수 있는 정수들이다.
길이나 나온 빈도 수 등이다.
일단 글자를 key로 빈도 수를 key 값으로 해두면, 길이는 key의 길이로 구해서 list에 넣으면 된다.
그리고 우선 순위에 맞게 정렬을 하면 된다.

일단 입력한 문자들을 한 리스트에 넣어 for 문을 돌려 M의 숫자보다 작으면 continue 해버린다.
M의 숫자보다 크면 딕셔너리에 넣어야하기 때문에 딕셔너리 key 값에 있는지 확인하고 넣는다.
(있으면 +1, 없으면 key값을 1로 새로 선언)

만든 딕셔너리를 다시 보며 list에 [글자, 글자 길이, 글자 빈도수] 순으로 넣는다.
그리고 글자 빈도수, 글자 길이, 글자 순으로 정렬을 해야하는데, 그냥 무턱대고 하면 안된다.
글자 빈도수, 글자 길이는 내림차순으로 정렬을 해야하는데, 글자는 사전 순이기 떄문에 오름차순이다.
그래서 글자 빈도수, 글자 길이는 정수이기 때문에 음수로 변환 후 다 같이 오름차순으로 정렬하면 된다.

최종 코드

N, M = map(int, input().split())

wd = {}
w_list = []

for _ in range(N) :
    w = input()
    if len(w) < M : continue
    else : 
        if w not in wd.keys() :
            wd[w] = 1
        else :
            wd[w] += 1
            
for x in wd.keys() :
    w_list.append([x, len(x), wd.get(x)])
    
sorted_w_list = sorted(w_list, key=lambda x: (-x[2], -x[1], x[0]))
            
    
for x in sorted_w_list :
    print(x[0])