[백준] 좌표 압축 (18870번)

문제

수직선 위에 N개의 좌표 X1, X2, ..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다.

Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표의 개수와 같아야 한다.

X1, X2, ..., XN에 좌표 압축을 적용한 결과 X'1, X'2, ..., X'N를 출력해보자.

입력

첫째 줄에 N이 주어진다.

둘째 줄에는 공백 한 칸으로 구분된 X1, X2, ..., XN이 주어진다.

출력

첫째 줄에 X'1, X'2, ..., X'N을 공백 한 칸으로 구분해서 출력한다.

제한

• 1 ≤ N ≤ 1,000,000
• -109 ≤ Xi ≤ 109

예제 입력 1 복사

5 2 4 -10 4 -9

예제 출력 1 복사

2 3 0 3 1

예제 입력 2 복사

6 1000 999 1000 999 1000 999

예제 출력 2 복사

1 0 1 0 1 0

---

입력으로 들어오는 수 들의 대소를 비교해 작은 순서대로 0, 1, 2, ... 을 부여해주면 된다.

일단 기존의 순서대로 좌표를 저장할 v 에 모든 값을 넣어놓은 후 

v를 v2 벡터에 복사한다. v2를 정렬해서 작은 수부터 읽는다.

각 수를 읽으면서 map에 순서를 기록한다. 0은 map이 비어있다는 뜻이 되므로 1부터 기록하자.

맵에 순서대로 key, value를 입력하면서 이미 입력한 수에 대해서는 continue로 넘어가고 입력하지 않은 수만 새로 입력해주자. 

입력이 끝나면 맨 처음 v 벡터를 순서대로 읽어서 해당하는 값의 인덱스가 몇번인지 맵에서 꺼내면 끝이다.

priority queue를 써도 구현이 가능할 것 같다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EC%A2%8C%ED%91%9C%20%EC%95%95%EC%B6%95%20(18870%EB%B2%88).cpp

chosh95/STUDY

 

C++ 코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
int N;
vector<int> v;
map<int, int> m;
int main()
{
	cin >> N;
	vector<int> v2;
	for (int tmp, i = 0; i < N; i++) {
		cin >> tmp;
		v.push_back(tmp);
	}

	v2 = v;
	sort(v2.begin(), v2.end());
	int idx = 1;
	for (int i = 0; i < N; i++) {
		if (m[v2[i]] == 0)
			m[v2[i]] = idx++;
		else continue;
	}

	for (int i = 0; i < N; i++) {
		cout << m[v[i]] - 1 << " ";
	}
}