[백준] 18870번: 좌표 압축 (JavaScript, NodeJS)

문제

수직선 위에 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

정답 풀이

const fs = require('fs');

// 백준 제출 할 때 주석 제거
// const readFileSyncAddress = '/dev/stdin';

// VSC 테스트 할 때 주석 제거
const readFileSyncAddress = 'input.txt';

let input = fs.readFileSync(readFileSyncAddress).toString().trim().split('\n');

// 문제 풀이
function solution(input) {
    [n, x] = input;
    x = x.split(' ').map(i => Number(i));
    
    const answer = [];
    const set = Array.from(new Set([...x])).sort((a, b) => a - b);
    const object = {};

    set.forEach((item, idx) => object[item] = idx);

    for (let i = 0; i < x.length; i++) {
        answer.push(object[x[i]]);
    }

    return answer.join(' ');
}

// 제출
console.log(solution(input));

 

맨 처음에는 findIndex 메소드를 사용했는데, 시간 초과가 떠서 object에 담아주는 방식으로 바꿨다.

findIndex는 시간복잡도가 o(n) 이지만, object로 처리하면 o(1)이 된다.