[백준 - python] 20920번: 영단어 암기는 괴로워

문제 링크

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

문제 풀이

from sys import stdin

input = stdin.readline

n, m = map(int, input().rstrip().split())
memorize = {}

for _ in range(n):
    word = input().rstrip()
    if len(word) >= m:
        if word in memorize:
            memorize[word][0] += 1
        else:
            memorize[word] = [1, len(word), word]

for word in sorted(memorize, key=lambda x:(-memorize[x][0], -memorize[x][1], memorize[x][2])):
    print(word)

20920 입출력 예시

*key point: 딕셔너리와 정렬 함수를 잘 활용하는 것이 가장 중요하다. 특히 정렬 함수를 쓸 때 람다식을 사용할 수 있으면 쉽게 풀이가 가능하다. 주어진 세 우선순위에 따른 정렬 기준이 람다식의 리턴값이 되도록한다. 이때 -를 곱하면 내림차순으로 정렬된다.