[코딩테스트]BJ11650 좌표 정렬하기 - C++

[문제]

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

[입력]

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

[출력]

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

[문제 풀이]

앞전에 먼저 풀었던 좌표 정렬하기2 와 같은 문제이다.
sort의 3번째 인자를 이용하여 오름차순, 내림차순을 원하는 방식으로 조작할 수 있어야한다.

[코드]

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

using namespace std; 
bool compare(const pair<int, int> &a, const pair<int, int> &b) {
	if (a.first == b.first)
		return a.second < b.second;
	else
		return a.first < b.first;
}
int main() {
	int N; 
	vector<pair<int, int>> dots;
	cin >> N; 

	for (int i = 0; i < N; i++) {
		int x, y;
		cin >> x >> y;
		dots.push_back(make_pair(x, y));
	}
	 
	sort(dots.begin(), dots.end(), compare);

	for (int i = 0; i < N; i++)
		cout << dots[i].first << " " << dots[i].second << '\n';

	return 0;
}