[코딩테스트] 백준 영단어 암기는 괴로워 파이썬(Python)

https://www.acmicpc.net/problem/20920

20920번: 영단어 암기는 괴로워

import sys

n, m = map(int, sys.stdin.readline().split())
words = {}
for _ in range(n):
  word = sys.stdin.readline().rstrip()
  if len(word) < m:
    continue
  if word not in words:
    words[word] = words.get(word, 0) + 1
  else:
    words[word] += 1

words = sorted(words.items(),key=lambda x : (-x[1], -len(x[0]), x[0]))

for word, count in words:
  print(word)

먼저 sys 모듈을 이용하여 입력값을 받아옵니다.

• input() 함수를 사용할 경우 시간초과가 나왔습니다.
• 반복문으로 n개의 단어를 입력받습니다.

이때 길이가 m이상인 경우에만 words라는 딕셔너리에 저장합니다.

words는 word를 key로 각 단어의 등장 횟수를 value로 갖도록 합니다.

• words.get(word, 0)은 딕셔너리 words에서 key가 word인 항목을 찾아 반환합니다.
• 만약 해당 key가 존재하지 않는다면 기본값으로 0을 반환하도록 하고 반환된 값에 1을 더하여 해당 단어의 등장 횟수를 기록합니다.

모든 단어의 등장 횟수를 계산하고 나면 words 딕셔너리를 정렬합니다.

• sorted() 함수의 key 옵션에 lambda함수를 사용하여 정렬을 합니다. 정렬 기준을 아래와 같습니다.

1. -x[1] : 등장 횟수를 내림차순으로 정렬
2. len(x[0]) : 단어 길이에 대하여 내림차순 정렬
3. x[0] : 단어의 사전순으로 오름차순 정렬

마지막으로 words에서 단어를 하나씩 출력합니다.