(C++) 백준 11650번 - 좌표 정렬하기

https://www.acmicpc.net/problem/11650

문제

> 2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력해라.
> 첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

접근

sort()함수를 사용하여 정렬하는데 세번째 파라미터를 정의해 문제에서 주어진 기준을 구현한다. x좌표순으로 정렬하는데 같을경우 y좌표를 기준으로 정렬한다.

문제해결

> 좌표 쌍으로 이루어졌으므로 pair로 점들을 입력받는다.
> pair쌍을 입력으로 받는 sort()의 세번째 매개변수로 들어갈 comp함수를 정의한다.
> 입력으로 들어온 두 좌표 a, b의 first 즉, x좌표가 같다면 a와 b의 second값인 y좌표를 비교한다. a가 더 작다면 true를 반환해 a를 b보다 앞에오게 정렬하고 false가 반환되면 a가 더 크다는 뜻이므로 b가 앞에오게 정렬한다.
> else는 두 x좌표가 다를 때의 경우이므로 둘의 x좌표를 비교해 true가 나오면 a가 앞 false가 나오면 b가 앞에 오게 정렬한다.
> 정렬된 좌표를 출력한다.

코드

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

bool comp(pair<int, int> a, pair<int, int> b)
{
	if (a.first == b.first)
		return a.second < b.second;
	else
		return a.first < b.first;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);

	int N;
	cin >> N;

	vector<pair<int, int>> Sort(N);
	for (int i = 0; i < N; i++) cin >> Sort[i].first >> Sort[i].second;

	sort(Sort.begin(), Sort.end(), comp);

	for (auto i : Sort) cout << i.first << " " << i.second << '\n';
}

후기

전에 브론즈단계에서 sort의 세번째 매개변수를 정의하여 단어정렬하는 문제를 푼적이 있다. 구현자체는 어렵지않았다.
이번 문제는 추가로 좌표, pair을 사용하여 쌍을 넣었을 때의 비교 및 정렬이다.