[18870] 좌표 압축

Byeongmin·2021년 6월 27일
0
post-thumbnail

[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

코드

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

using namespace std;

int main() {
    /* 빠른 cin cout */
    ios_base :: sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    /* 입력 */
    int N;
    cin >> N;
    vector<int> input, position;

    /* 풀이 */
    int tmp;
    for(int i = 0; i < N; i++) {
        cin >> tmp;
        input.push_back(tmp);
        position.push_back(tmp);
    }

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

    /* 출력 */
    for(int i = 0; i < N; i++) {
        cout << lower_bound(position.begin(), position.end(), input[i]) - position.begin() << ' ';
    }
}

부가 설명

두개의 vector를 만들어 position이라는 vector는 정렬을 하고 중복을 제거했고,
각 input에 대해 lower bound를 사용해 position에서 찾은 index와 0부터 시작한 순서가 같으므로 index를 출력하도록 했다.

  • 예시
    예시

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

profile
Handong Global Univ.

0개의 댓글