[Seop's의 코드풀이] 백준 2910 빈도정렬 - Python

이번에 풀어볼 문제는 백준 2910번인 빈도 정렬 문제이다.

문제는 다음과 같다.

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

2910번: 빈도 정렬

들어온 순서와 빈도를 고려하기 때문에

요전에 풀었던 1713번과 매우 유사한 방법으로 해결 할 수 있다.

 

하나의 딕셔너리를 생성 한 후 다음과 같이 적재한다면 어렵지 않게 해결 할 수 있다.

 

Key : 메세지

Value : [빈도, 들어온 순서]

 

딕셔너리에 전부 적재 했다면 정렬을 통해 진행 답을 도출해 내면 끝이다.

전체적인 코드는 다음과 같다.

import sys
input = sys.stdin.readline

if __name__ == "__main__":
    N, C = map(int, input().split(" "))
    num = list(map(int, input().split(" ")) )
    num_dict = dict()
    for i in range(N) :
        if num[i] in num_dict :
            num_dict[num[i]][0] += 1
        else :
            num_dict[num[i]] = [1, i]
    
    result = sorted(num_dict.items(), key= lambda x: (-x[1][0], x[1][1] ))
    answer = []
    for i in result :
        for j in range(i[1][0]) :
            answer.append(str(i[0]))
    print(" ".join(answer))