[BOJ][C++] 백준 18870번: 좌표 압축

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

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. 받은 값들을 오름차순으로 정렬한 후 중복을 제거한다
2. 정렬한 값에 수를 할당한다 (자료구조 map 사용)
3. 받은 값들을 순회하며 할당된 값을 출력한다

 

예를들어, 예제 1인 위에 상황에서

2 4 -10 4 -9

 이걸 오름차순으로 정렬하고 중복을 제거하면 {-10, -9, 2, 4} 이다.

순서대로 0부터 3까지의 값을 할당하면 아래처럼 된다.

map = [{-10, 0}, {-9, 1}, {2, 2}, {4, 4}]

 

이제 다시 원래 입력값 {2, 4, -10, 4, -9} 를 돌며 할당된 값을 출력하면

{2, 4, 0, 4, 1}

 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

int main() {
    int n;
    cin >> n;
    vector<int> input (n);
    vector<int> sortedInput(n);
    for(int i=0; i<n; i++) {
        cin >> input[i];
        sortedInput[i] = input[i];
    }
    
    map<int, int> m;
    int cur = 0;
    sort(sortedInput.begin(), sortedInput.end());
    sortedInput.erase(unique(sortedInput.begin(), sortedInput.end()), sortedInput.end());
    for(int i : sortedInput) {
        m.insert({i,cur});
        cur++;
    }
    vector<int> answer;
    for(int i : input) {
        answer.push_back(m[i]);
    }
    
    for(int i : answer) {
        cout << i << " ";
    }
    
    return 0;
}