[C++] 백준(BOJ) - 2910 : 빈도 정렬
인프런에 있는 큰돌님의 강의 10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트를 듣고 정리한 필기입니다.
문제
위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.
창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.
만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.
이렇게 정렬하는 방법을 빈도 정렬이라고 한다.
수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.
입력
첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)
둘째 줄에 메시지 수열이 주어진다.
출력
첫째 줄에 입력으로 주어진 수열을 빈도 정렬한 다음 출력한다.
예제 입력
5 2
2 1 2 1 2
예제 출력
2 2 2 1 1
알고리즘
- 자료구조
- 정렬
- 해시를 사용한 집합과 맵
풀이
- 입력으로 주어진 수열을 담을 map과 순서를 담을 map 두가지를 선언
- 순서를 담을 map의 경우 map에 요소가 있는지 없는지 검사를 한 후 순서를 담는 과정을 추가적으로 코딩
- map을 vector<pair<int,int» 형에 담아 second를 기준으로 커스텀 정렬
- 커스텀 정렬 함수 선언 및 정의
- 정렬된 vector를 순회하며 vector의 요소의 second(개수)만큼 first(값)을 출력
코드
// BOJ - 2910 : 빈도 정렬
#include <bits/stdc++.h>
using namespace std;
int n, c;
map<int, int> order; // 들어온 순서를 기록하는 맵
bool cmp(pair<int, int> &a, pair<int, int> &b)
{
if (a.second == b.second) return order[a.first] < order[b.first]; // 개수가 같으면 들어온 순서대로 정렬
return a.second > b.second; // 많은 수 순서대로 정렬
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n >> c;
map<int, int> list; // 입력을 받을 맵
for (int i = 0; i < n; i++)
{
int a;
cin >> a;
list[a]++;
if (order[a] == 0) order[a] = i + 1; // 먼저 들어온 순으로 order[i].second 에 순번이 기록 된다
}
vector<pair<int, int>> vec(list.begin(), list.end()); // map을 들어온 순서대로 정렬하기 위해 vector<pair<int,int>> 형을 선언하고 list 맵으로 초기화 할당
sort(vec.begin(), vec.end(), cmp); // 커스텀 비교함수를 이용해서 map의 second 기준으로 정렬
for (auto x : vec)
{
for (int i = 0; i < x.second; i++)
{
cout << x.first << ' ';
}
}
return 0;
}
평가
- map을 이용해서 빈도수 정렬을 할 때에는 vector를 이용해서 커스텀 정렬을 하는 방법을 이용한다
- map은 참조만해도 요소가 생기는 성질을 파악하고 key 값이 없을 때 값을 할당해주는 로직을 작성한다
- 정렬을 위한 커스텀 함수를 자유자재로 만들 수 있다
