백준2910 문제 보러가기

문제

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

창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 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

예제 입력 2

9 3
1 3 3 3 2 2 2 1 1

예제 출력 2

1 1 1 3 3 3 2 2 2

예제 입력 3

9 77
11 33 11 77 54 11 25 25 33

예제 출력 3

11 11 11 33 33 25 25 77 54

문제 해결 과정

처음에는 딕셔너리에 빈도 수와 먼저 나온 것에 대한 정보(?)를 넣어서 정렬하려고 했다.
하지만 먼가 다른 방법이 있을 것 같아서 sort, sorted 함수를 이용했다.
key에다가 빈도 수에 대한 정보를 넣으면 정렬이 되지 않을까 하고 생각했다. 그리고 빈도 수가 같을 경우 아무런 조치를 하지 않았을 때에 들어온 순으로 정렬이 되기 때문에 빈도 수에 대한 정보만 넣으면 된다고 생각했다.

sorted(num_list, key = lambda x : num_list.count(x))

내가 생각한 방식은 이것이었다.
그런데 이게 빈도 수가 모두 다르다면 잘 작동한다.
예를 들어, 입력이 2 1 2 1 2 라면, 출력했을 때 2 2 2 1 1로 잘 출력이 된다.
하지만, 입력이 3 2 3 2 1 이면 3 2 3 2 1 그대로 출력이 된다.
그러니까 빈도 수가 같은 한 쌍이라도 존재하면 위와 같이 출력이 된다.

결국엔 딕셔너리에 빈도 수와 먼저 들어온 것에 대한 정보를 넣어서 저장한 뒤 빈도 수, 먼저 들어온 것 순서로 정렬했다.

최종 코드

N, C = map(int, input().split())
num_list = list(map(int, input().split()))
dict_info = {}
idx = 1
answer = ''
for num in num_list :
    if num in dict_info :
        dict_info[num][0] += 1
    else :
        dict_info[num] = [1, idx]
        idx += 1
        
sorted_num = sorted(dict_info.items(), key = lambda x: (-x[1][0], x[1][1]) )
for key, value in sorted_num :
    answer += (str(key) + ' ') * value[0]
    
print(answer[:-1])