[백준 20920번] 파이썬 - 영단어 암기는 괴로워

백준 20920_영단어 암기는 괴로워

시간 제한 1초 (추가 시간 없음), 메모리 제한 1024MB

# 조건

• 화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다.
• 그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다.
• 화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.
  1. 자주 나오는 단어일수록 앞에 배치한다.
  2. 해당 단어의 길이가 길수록 앞에 배치한다.
  3. 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치한다
• M보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가 M이상인 단어들만 외운다고 한다.
• 화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.

입력

• 첫째 줄에는 영어 지문에 나오는 단어의 개수 N과 외울 단어의 길이 기준이 되는 M이 공백으로 구분되어 주어진다. (1≤N≤100000, 1<=M<=10)
• 둘째 줄부터 N+1번째 줄까지 외울 단어를 입력받는다.
  • 이때의 입력은 알파벳 소문자로만 주어지며 단어의 길이는 10을 넘지 않는다.
• 단어장에 단어가 반드시 1개 이상 존재하는 입력만 주어진다.

출력

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

# 접근 방법

• Dictionary와 lambda를 잘 활용해야 하는 문자열 문제이다.
• 우선 등장 횟수에 따라 단어를 기록해준 count 딕셔너리 1개
  • { 등장 횟수(int) : [ 등장 단어 리스트 ] }
• 단어마다 등장 횟수 기록해준 words 딕셔너리 1개를 사용해주었다.
  • 조건문을 최소화하기 위하여 defaultdict를 이용하여 주었다.
  • { 등장 단어 리스트 : 등장 횟수 }
  • 입력을 받으며 M 이상의 길이 단어가 등장하면
  • words[단어] += 1 을 해주었고
• 입력이 끝난 다면 words 딕셔너리를 순회하며, count[words[단어]].append() 를 통하여 기록해주었다.
  • ex) apple 이 3번 등장한다면 count = {1:[], 2:[], 3:['apple']} 이 된다.
• N부터 0까지 역순으로 순회하며 sort 후 출력해준다.
  • sort의 조건은 lambda를 이용하는데
  • 주어진 우선 순위중 1번 -> 등장 횟수는 만족하였으므로
  • 길이와, 사전 순을 조건으로 해준다.
• 길이는 내림차순, 사전은 오름차순이 되어야 하므로, 길이에 -를 주면서 내림차순으로 만들어 주었다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  
from collections import defaultdict  


count = {i:[] for i in range(1, 100001)}  
words = defaultdict(int)  
N, M = map(int, input().split())  
for _ in range(N):  
    word = input().rstrip()  
    if len(word) < M:  
        continue  
    else:  
        words[word] += 1  

for i in words:  
    count[words[i]].append(i)  

for j in range(N, 0, -1):  
    if count[j]:  
        count[j].sort(key=lambda x:(-len(x), x))  
        print(*count[j], sep='\n')