[백준] 18870번 좌표 압축 - C++

좌표압축

시간 제한	메모리 제한
2 초	512 MB

---

문제

수직선 위에 N개의 좌표 X1, X2,..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다.

Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표 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	예제 출력 1
5 2 4 -10 4 -9	2 3 0 3 1

---

예제 입력2	예제 출력 2
6 1000 999 1000 999 1000 999	1 0 1 0 1 0

---

알고리즘 분류

• 정렬
• 값 / 좌표 압축

---

풀이 방법

1. n 입력
2. 길이 n의 포인터 배열 nums 생성
3. 수 n개 입력
4. 포인터 배열 sorted 생성 후 nums 복사
5. sorted의 원소(주소)를 참조값 기준으로 정렬
6. 참조값에 정렬된 순서에 맞게 새로운 값 부여
7. 원래의 배열 nums로 입력 순서에 맞게 새로운 값 출력

코드

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

int main()
{
	// 입출력을 빠르게 하는 함수
	std::cin.tie(NULL);
	std::ios::sync_with_stdio(false);

	int n;
	std::cin >> n;

	std::vector<int*> nums(n);
	for (int i = 0; i < n; ++i)
	{
		nums[i] = new int;
		std::cin >> *nums[i];
	}

	std::vector<int*> sorted = nums;
    
    // 람다 객체에 정렬 기준을 정의해서 전달
	std::sort(sorted.begin(), sorted.end(), [](int* a, int* b) { return *a < *b; });
    
    // 참조값을 새로운 값으로 업데이트
	int order = 0;
	int prev = 1000000001;
	for (int* num : sorted)
	{
		if (*num != prev)
		{
			prev = *num;
			*num = order++;
		}
		else
		{
			*num = order - 1;
		}
	}

	// 입력순서에 맞게 출력
	for (int* num : nums)
	{
		std::cout << *num << ' ';
		delete num;
	}

	return 0;
}

마치며

 처음엔 입력순서와 출력순서를 맞추려고 for문을 이용했는데 시간초과가 나서 고민하던 중에 포인터 배열을 두 개 쓰면 한 개의 데이터 집합을 두 가지 방법으로 관리할 수 있겠다 싶어서 포인터 배열을 이용해서 해결해 봤습니다. 제가 사용한 방법 이외에도 map을 이용하는 방법 등 여러 가지 방법이 있습니다.