[C++] 백준 2910번:빈도 정렬

문제링크

2910번: 빈도 정렬

문제

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

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

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

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

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

입력

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

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

출력

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

 

풀이

위에서는 index를 수로 하는 배열 2개 (크기가 1001인)를 만들었다. 하나는 빈도 수를 담는 배열, 다른 하나는 입력 받은 순서를 담는 배열이다. 최대로 들어올 수 있는 수가 10억이기 때문에 10억 크기의 배열을 만드는 것은 불가능기 때문에 자료구조 map을 사용했다.

 

각 맵의 첫 번째 요소는 입력받은 숫자를 의미한다.

하나는 빈도 수를 담는 맵, 다른 하나는 입력 받은 순서를 담는 맵이다.

 

만약 숫자 3에 대해 빈도 수와 입력 받은 순서를 알고 싶다면

각각 fre[3], order[3] 과 같은 방식으로 접근하면 된다.

 

맵의 크기만큼 돌면서 빈도 수가 가장 많은 index를 찾는다. 이 때 빈도 수가 같다면 입력 받은 순서를 비교하여 갱신해주고 빈도 수만큼 출력해준다.

 

코드

#include <iostream>
#include <map>
using namespace std;
 
int n, c, val;
map<int, int> fre;
map<int, int> order;
 
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
 
    cin >> n >> c;
 
    int num = 1;
 
    // make order
    for (int i = 0; i < n; i++) {
        cin >> val;
        fre[val]++;
        
        if (order[val] == 0) {
            order[val] = num++;
        }
    }
 
    while (true) {
        int maxFre = 0;    // max frequency of number
        int index = 0;    // the index of frequency max
 
        // find
        for (int i = 1; i <= fre.size(); i++) {
            if (maxFre < fre[i]) {
                maxFre = fre[i];
                index = i;
            }
            // if same of frequency
            else if (maxFre == fre[i]) {
                if (order[index] > order[i]) {
                    index = i;
                }
            }
        }
 
        if (maxFre == 0) {
            break;
        }
 
        // make zero
        while (fre[index]--) {
            cout << index << ' ';
        }
    }
 
}