[Python] 백준 2910번

2910. 빈도 정렬 (Silver 3)

# 자료구조

위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.

창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.

만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.

이렇게 정렬하는 방법을 빈도 정렬이라고 한다.

수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오

 

입력

첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)

둘째 줄에 메시지 수열이 주어진다.

 

출력

첫째 줄에 입력으로 주어진 수열을 빈도 정렬한 다음 출력한다.

 

예제 입력1

5 2
2 1 2 1 2

예제 출력1

2 2 2 1 1

 

 

 

풀이)

메시지의 길이 n과 c보다 작은 숫자를 입력해야할 때 필요한 c를 각각 입력받았다.

code를 선언해서 리스트 형태로 암호를 입력받게 했다.

그 다음 dic라는 이름의 딕셔너리를 선언해주었다.

for문을 사용해서 code에 있는 원소들을 검사하며 dic에 있지 않은 경우에는 해당 원소에 해당하는 인덱스의 value를 0으로 저장해주었고, 그 외의 경우에는 해당 원소의 인덱스의 value값을 1 증가시켜주었다.

이 과정을 거친 다음, sorted를 사용해서 dic원소에서 value값이 큰 순으로(내림차순) 정렬한 값을 다시 저장해주었다.

마지막으로 dic안에 있는 key와 value에 대해서 value의 길이 만큼 반복할 때, key값만을 공백을 사이에 두고 출력하게 해주었다.

import sys
input = sys.stdin.readline

n, c = map(int, input().split())   # n, c 공백을 두고 입력
code = list(map(int, input().split()))  # 암호 입력

dic = {} # 딕셔너리

for i in code:
    if i not in dic:   # 딕셔너리에 없으면 
        dic[i] = 0     # 해당 숫자에 해당하는 key의 value 0으로 설정
    dic[i] += 1        # 딕셔너리에 있으면 value값 1 증가

dic = sorted(dic.items(), key=lambda x: x[1], reverse=True)  # value값이 큰 것부터 정렬

for key, value in dic:
    for i in range(value):
        print(str(key), end=" ")  # key값만 출력