[백준 알고리즘] 11651번: 좌표 정렬하기 2

11651번: 좌표 정렬하기 2

문제
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;

}

Written by@전여훈 (Click Me!)
고민이 담긴 코드를 만들자, 고민하기 위해 공부하자.