[백준/C++] 11651번: 좌표 정렬하기2

문제

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

입력

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

출력

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

 

풀이

comp 함수를 지정하여 주어진 조건대로 비교한다.

1. y좌표가 다르면 y좌표가 증가하는 순으로 정렬

2. y좌표가 같으면 x좌표가 증가하는 순으로 정렬

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> ci;

bool comp(const ci &a, const ci &b){
    if (a.second != b.second)
        return a.second < b.second;
    return a.first < b.first;
}

int main(){
    int n;
    vector<ci> grid;

    // 입력
    cin >> n;
    grid.assign(n, {0, 0});
    for (int i=0; i<n; i++)
        cin >> grid[i].first >> grid[i].second;

    // 연산
    sort(grid.begin(), grid.end(), comp);

    // 출력
    for (int i=0; i<n; i++)
        cout << grid[i].first << " " << grid[i].second << "\n";
}