BOJ/정렬

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

JWonK 2021. 7. 31. 16:24
728x90
반응형

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

 

2910번: 빈도 정렬

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

www.acmicpc.net

문제

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

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

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

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

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

입력

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

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

출력

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

 

빈도수 정렬을 하기 위해 등장했는지 안했는지 확인하는 배열을 선언해서 해결하려했지만 C의 최대값 범위가 

배열의 범위를 초과하기 때문에 다른 방법을 사용해야겠다.

 

그래서 map이라는 C++의 STL을 사용하였고, map을 통해 등장했는지 안했는지 확인하고, 그 빈도수를 value값 증가로 해결해주었다.

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <cmath>
#define INF 987654321
using namespace std;

typedef long long ll;
// BOJ :: https://www.acmicpc.net/problem/7795

bool compare(pair<int,pair<int,int>> a, pair<int,pair<int,int>> b) {
	if (a.second.second == b.second.second)
		return a.first < b.first;
	else
		return a.second.second > b.second.second;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);

	int N, C;
	cin >> N >> C;

	vector<pair<int,pair<int, int>>> v;
	map<int, int> m;
	queue<pair<int,int>> q;

	int ps = 1;
	for (int i = 0; i < N; i++) {
		int x; cin >> x;
		if (x <= C) {
			if (m.find(x) == m.end()) {
				m.insert({ x, 1 });
				q.push({ ps, x });
				ps++;
			}
			else {
				m[x]++;
			}
		}
	}

	while (!q.empty()) {
		int fast = q.front().first;
		int data = q.front().second;
		q.pop();

		v.push_back({ fast, {data,m[data] } });
	}

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

	for (int i = 0; i < v.size(); i++) {
		int A = v[i].first;
		int B = v[i].second.first;
		int C = v[i].second.second;
		for (int j = 0; j < C; j++) {
			cout << B << " ";
		}
	}
	return 0;
}
728x90
반응형