BOJ 18870 좌표 압축

www.acmicpc.net/problem/18870

18870번: 좌표 압축

좌표 압축 문제는 테크닉은 어렵지 않지만 더 어려운 문제에서 활용되어 쓰인다고 한다.

처음에는 pair를 사용해서 인덱스와 값을 입력받고 정렬 후 O(N)으로 처리하려 했는데 잘 안돼서 찾아보니

unique와 lower_bound를 사용해서 해결하는 문제였다.

 

우선 배열 2개를 선언해서 같은 값을 받아준다.

	vector<int> ar(n), ans(n);
	for (int i = 0; i < n; i++) {
		cin >> ar[i]; ans[i] = ar[i];
	}

이 후에 정렬을 하고 중복을 제거해주면 이런 상태가 된다.

 

 

  2   4   -10   4   -9

-10  -9    2    4    

 

이 상태에서 lower_bound를 통해 인덱스를 구해주면 된다.

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

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

	int n;
	cin >> n;
	vector<int> ar(n), ans(n);
	for (int i = 0; i < n; i++) {
		cin >> ar[i]; ans[i] = ar[i];
	}
	sort(ans.begin(), ans.end());
	auto it = unique(ans.begin(), ans.end());

	for (int i = 0; i < n; i++)
		cout << lower_bound(ans.begin(), it, ar[i]) - ans.begin() << ' ';

	return 0;
}

해답을 안 봤으면 못 떠올렸을 것 같은 문제였다..