백준 2910 문제풀이

문제 링크

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

문제 설명

위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다. 창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다. 만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다. 이렇게 정렬하는 방법을 빈도 정렬이라고 한다. 수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.

문제 분석

문제에서 바로 힌트가 주어진다. x,y가 있을 때 많이 등장하는 수가 앞으로 오는 순서로 정렬을 시도한다. pair<int,int> pv를 활용하여 pair<숫자, 빈도수> 를 기준으로 정렬을 하는 것으로 기준점을 잡았다. 따라서 정렬을 할 수 있는 cmp함수를 만드는 것으로 문제를 해결한다.

코드

#include<bits/stdc++.h>
using namespace std;
int n,c,a[1004];
map<int,int> mp, mp_first;
vector<pair<int,int>> pv;

bool cmp(pair<int,int> a, pair<int,int> b){
    if(a.second == b.second){
        return mp_first[a.first] < mp_first[b.first];
    }
    return a.second > b.second;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL), cout.tie(NULL);
    cin >> n >> c;
    for(int i = 0; i < n; i++){
        cin >> a[i];
        mp[a[i]]++;

        if(mp_first[a[i]] == 0){
            mp_first[a[i]] = i + 1;
        }
    }
    for(auto it : mp){
        pv.push_back({it.first, it.second});
    }
    sort(pv.begin(), pv.end(), cmp);

    for(auto i : pv){
        for(int j = 0; j < i.second; j++){
            cout << i.first << ' ';
        }
    }

    return 0;
}

a.second == b.second를 보면 빈도수를 기준으로 빈도수가 같다면 첫 번째 등장 위치를 기반으로 정렬을 한다. 아니라면 빈도수가 큰 것을 기준으로 정렬을 한다.

Reference

https://en.wikipedia.org/wiki/Quadtree