[210718][백준/BOJ] 18870번 좌표 압축

문제

입출력

풀이

배열 내에서 자신보다 작은 값이 몇개 존재하는지를 찾는 문제이다.
이중 for문을 사용하면 될거같았지만 n의 값이 최대 1백만이여서 O(n^2)으로는 문제를 풀 수가 없다. 따라서 O(nlogn) 이하의 시간복잡도로 해결해야 한다.

1. 이중 벡터를 선언하고 배열과 자리값을 넣는다. (2,0), (4,1), (-10,2), (4,3), (-9,4)

2. 오름차순으로 정렬해준다. (-10,2), (-9,4), (2,0), (4,1), (4,3)

3. 자신보다 작은값이 몇개 존재하는지 찾기 위한 배열을 하나 선언하고 arr[num[0].second]을 0으로 초기화해준다. arr[num[0].second]는 arr[2]와 같으며 -10보다 작은값은 존재하지 않기 때문에 0으로 초기화해준다.

4. 반복문을 돌면서 (i=1~n) num[i]과 num[i-1]의 first 값이 같은지 비교해서 같다면 이전 값을 그대로 넣어주고 같지않다면 이전 값+1을 넣어준다.

코드

#include <bits/stdc++.h>
using namespace std;

int arr[1000002];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);

	int n;
	cin >> n;
	vector<pair<int, int>> num(n);

	for (int i = 0; i < n; ++i)
	{
		cin >> num[i].first;
		num[i].second = i;
	}

	sort(num.begin(), num.end());
	arr[num[0].second] = 0;

	for (int i = 1; i < n; ++i)
	{
		if (num[i].first == num[i - 1].first)
			arr[num[i].second] = arr[num[i - 1].second];
		else
			arr[num[i].second] = arr[num[i - 1].second] + 1;
	}
	for (int i = 0; i < n; ++i)
		cout << arr[i] << ' ';
}