백준 11651 좌표 정렬하기 2 C++

문제

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

입력

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

출력

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

 

내 풀이

sort함수를 이용하여 정렬한다.

compare함수를 통해 어떻게 정렬할 것인지에 대한 기준을 설정한다.

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

using namespace std;

bool compare(const pair<int, int> &a, const pair<int, int> &b) {
	if (a.second != b.second)
		return (a.second < b.second);
	else
		return (a.first < b.first);
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);

	std::vector <pair <int, int>> coordinate;

	int n;
	int x, y;

	cin >> n;

	for (int i = 0; i < n; i++) {
		cin >> x >> y;
		coordinate.push_back(make_pair(x, y));
	}

	sort(coordinate.begin(), coordinate.end(), compare);
	for (int i = 0; i < n; i++)
	{
		cout << coordinate[i].first << " "
			<< coordinate[i].second << '\n';
	}

	return 0;
}