BOJ 11651 - 좌표 정렬하기 2

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

풀이

이번에는 y를 먼저, x를 나중에 비교해야 한다는 점만 11650과 다르다. 좌표를 저장한 뒤 비교 함수의 우선순위만 (y, x)로 바꾸면 코드 구조 자체는 거의 동일하다.

즉 핵심은 정렬 알고리즘이 아니라 comparator다. 무엇을 먼저 비교해야 하는지 한 번 정리되면 나머지는 그대로 따라간다.

코드

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
class coordinate {
public:
	int x;
	int y;
	coordinate(int x, int y) : x(x), y(y) { }
};
 
bool compare(coordinate a, coordinate b) {
	if (a.y == b.y)
		return a.x < b.x;
	else
		return a.y < b.y;
}
 
int main() {
 
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
 
	int N;
	cin >> N;
 
	vector<coordinate> coord;
	int a, b;
	for (int i = 0; i < N; i++) {
		cin >> a >> b;
		coord.push_back(coordinate(a, b));
	}
 
	sort(coord.begin(), coord.end(), compare);
 
	for (int j = 0; j < N; j++)
		cout << coord[j].x << " " << coord[j].y << '\n';
	return 0;
}

복잡도

시간 복잡도: 좌표를 정렬하므로 $O(N \log N)$ 이다.
공간 복잡도: 좌표 배열을 저장하므로 $O(N)$ 이다.

마무리

좌표 정렬의 핵심은 어느 축을 먼저 비교하느냐에 있다. 이번에는 y를 우선 기준으로 삼고, 같은 높이에 있는 점만 x로 정리하면 된다.

문제#

입력#

출력#

풀이#

코드#

복잡도#

마무리#

문제

입력

출력

풀이

코드

복잡도

마무리