[algorithm] 백준 C++ 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

문제해결

1. 입력한 숫자를 v 백터에 저장 <- 정렬될 백터

2. v백터를 복사하여 v1 백터를 생성 <- 원본 백터

3. v 백터 정렬후 같은 숫자 삭제

4. lower_bound()함수를 이용하여 ( v1 백터(원본)의 원소 위치 - v.begin() )를 구함.

 

* 2, 4, -10, 4, -9 -> 정렬  -> -10, -9, 2, 4, 4 -> 같은 숫자 삭제 -> -10, -9, 2, 4

문제 조건에서 xi > xj 를 만족하는 원소의 개수는 정렬 된 원소의 위치를 구하면 되는 문제

* 위치값을 찾기 위해 find함수를 사용하면 시간 초과되는 문제가 있으니 lower_bound 함수를 사용

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main(void){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    int number;
    vector<int> v;

    cin >> n;
    for(int i = 0;i<n;i++){
        cin >> number;
        v.push_back(number);
    }
    vector<int> v1(v);

    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()),v.end());

    for(int i = 0; i<v1.size();i++ ){
        cout << lower_bound(v.begin(), v.end(), v1[i]) - v.begin() << ' '; // find 대신 lower bound
    }
    return 0;
}