6. 백준 20920 (실버3) : 영단어 암기는 괴로워 _ python풀이

문제

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

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

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

입력

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

---

*문제해설

  이 문제는 '자료구조' 문제다.

단어들을 입력받고 주어진 글자수 (M) 이하의 단어들을 희은이가 저장한 순서대로 출력하는 문제다. 입력받기 위해 '딕셔너리' 자료형을 사용하는게 올바를 것이다. 출력하기 위해 '단어'를 저장하고, 비교를 위해 '글자수'를 저장해야하기 때문이다. 또한 단어가 중복해서 나오는 경우가 있기 때문에 중복된 개수 표기도 딕셔너리로 함께 해주면 될 것이다.

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

희은이가 저장하는 기준은 위와 같다. 이런 요구사항을 다 받아들이기 위해서는 저 요소들을 딕셔너리에 다 저장하고 요소별로 정렬해주어야 할 것이다.

 

*나의코드

import sys
input = sys.stdin.readline

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

dic = {}
#        ([중복개수][단어길이][단어명])
# [단어명]

for _ in range(N):
    word = input().strip()
    if len(word) < M:
       continue
    if dic.get(word): #만약 dic에 word가 있으면 +1 (중복처리)
        dic[word][0] += 1
    else:
        dic[word] = [1, len(word), word] # 개수, 길이, 단어
        
#개수, 길이 는 내림차순 ('-'를 붙여서) 단어는 오름차순으로 정렬
result = sorted(dic.items(), key= lambda x: (-x[1][0], -x[1][1], x[1][2]))
        
for i in result:
    print(i[0])

 

---

*중요한 내용

1. word = input().strip() 부분에서 strip를 하지않게 된다면 '출력 형식이 잘못되었습니다' 라고 나온다. 그것을 유의해야한다.

2. sorted() 함수는 그냥 사용하면 단순 오름차순 정렬이 되지만 

sort(key= lambda x: (-기준1, 기준2, 기준3))

이렇게 사용하여 기준별로 정렬을 할 수 있다.