(C++) 백준 18870번 - 좌표 압축

용꿀 2023. 5. 8. 21:25

문제 링크 : https://www.acmicpc.net/problem/18870

18870번: 좌표 압축

수직선 위에 N개의 좌표 X1, X2, ..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다. Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표의 개수와 같아야 한다. X1, X2, ..., XN에 좌

www.acmicpc.net

풀이

먼저 입력받은 배열을 sort() 메서드를 사용해서 정렬하고, 그 후 unique()와 erase() 메서드를 사용해서 중복을 제거한다.

마지막으로 lower_bound() 메서드를 사용해서 현재 어느 위치에 존재하는지 알아낸다.

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

using namespace std;

vector<int> X;
int ax[1000005];
int n;

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

    cin >> n;
    for(int i = 0; i < n; i++){
        cin >> ax[i];
        X.push_back(ax[i]);
    }
    sort(X.begin(), X.end());
    X.erase(unique(X.begin(), X.end()), X.end()); // 중복 제거

    for(int i = 0; i < n; i++){
        cout << lower_bound(X.begin(), X.end(), ax[i]) - X.begin() << " ";
    }
}