May 05, 2020
문제
2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.
출력
첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.기존 정렬하기 코드에서 sort 함수의 인자로 새로 정의한 비교함수를 넣어주었다. 이 함수에서는 pair 의 second 끼리 비교하고 second가 서로 같으면 first끼리 다시 비교해서 first가 더 작은 수를 앞으로 정렬시키도록 구성했다.
#include <iostream>
#include <vector>
#include <utility>
#include <algorithm>
using namespace std;
bool cmp (const pair<int, int> &a, const pair<int, int>&b){
if(a.second < b.second) return true;
else if (a.second == b.second) return a.first < a.second;
return false;
}
int main (){
ios_base:: sync_with_stdio(false);
cin.tie(NULL);
int N;
cin >> N;
vector<pair<int, int> > v (N); // 페어벡터 만들고 최댓값으로 초기화
for (int i = 0 ; i < N ; i++){
int x, y;
cin >> x >> y;
v[i].first = x;
v[i].second = y;
}
sort(v.begin(), v.end(), cmp);
for (int i = 0 ; i < N ; i++){
cout << v[i].first << " " << v[i].second << "\n";
}
return 0;
}