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

빈도 정렬

문제

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

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

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

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

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

입력

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

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

출력

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

예제

문제 정의

이 문제에서는 각 숫자의 빈도수를 담는 map, 우선순위를 담는 map, 커스텀 정렬 함수를 만드는 것이 중요하다.

처음에 봤을 때 어떻게 풀지 막막했는데 계속 보고.. 여러 자료를 참고해서 겨우 풀었다 아마 나중에 다시 또 풀어봐야겠다

먼저 나는 입력받은 숫자의 빈도수를 mp 라는 변수명의 map에 저장했다.
그리고 동시에 mp[입력받은 값] == 1 일 경우 숫자의 우선순위를 담는 mp_first라는 변수명의 map에 입력받은 순서(인덱스+1) 를 저장했다.

그 후 mp의 각 요소들을 vector<pair<int, int>> v 에 저장한 후 커스텀 빈도 정렬 함수를 통해 정렬했다.

mp의 값을 벡터에 저장한 후 벡터 정렬에서 커스텀 함수(sorting)를 사용해 빈도 정렬했다.

빈도수(x.first, y.first)가 같은 경우에는 해당 숫자(x.second, y.second)의 우선순위(먼저나온)가 높은 pair가 앞으로 오도록 했다.

빈도수(x.first, y.first)가 같지 않다면 빈도수가 큰 pair가 앞으로 오도록 했다.

전체 코드는 다음과 같다.

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

bool sorting(pair<int, int> x, pair<int, int> y) {
  if (x.first == y.first) {
    return mp_first[x.second] < mp_first[y.second];
  }
  return x.first > y.first;
}
int main() {
  cin >> n >> c;
  for (int i = 0; i < n; i++) {
    cin >> a;
    mp[a]++;
    if (mp[a] == 1) {
      mp_first[a] = i + 1;
    }
  }

  for (auto i : mp) {
    v.push_back({i.second, i.first});
  }

  sort(v.begin(), v.end(), sorting);

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