CS/Algorithm & Data Structure

[BOJ][C++] 백준 11650 좌표 정렬하기

Yeji Heo 2022. 11. 20. 15:51

sort함수에 compare함수로 정렬기준을 만들어 사용.

좌표는 vector의 pair로 관리.

compare에서는 p1의first와 p2의 first(x좌표)가 같으면 second(y좌표)로 비교하도록 했다.

 

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

bool compare(pair<int, int> p1, pair<int, int> p2) {
	if (p1.first == p2.first) {
		return p1.second < p2.second;
	}
	else {
		return p1.first < p2.first;
	}
}

int main()
{
	int N;
	cin >> N;
	vector<pair<int, int>> v(N);

	for (int i = 0; i < N; i++) {
		cin >> v[i].first >> v[i].second;
	}

	sort(v.begin(), v.end(), compare);

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