[C/C++] 좌표 정렬하기 (백준 11650)

문제

 

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

 

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

 

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

 

입력

 

5
3 4
1 1
1 -1
2 2
3 3

 

출력

 

1 -1
1 1
2 2
3 3
3 4

 

풀이

 

정렬 부분을 공부하면서 다양한 방법으로 문제를 풀게되서 굉장히 새롭게 느껴진다. 일단 이번에는 정렬이지만 그냥 정렬이 아니라는 점에서 확실하게 하고 가야한다. 그래서 일단 pair를 통해서 좌표를 엮어 주었다.

 

그 이후 바로 sort를 진행했다. 그렇게 되면 자동으로 first를 먼저 정렬을 하고 second를 정렬하는 방식이 된다. 중간에서 이렇게 그냥 끝나서 허무했지만 만약 반대로 두 수가 반대가 되고 싶다면 sort에 옵션을 추가하면 될 것 같다.

 

코드

 

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>

using namespace std;

int main(void){
    vector<pair<int, int>> location;

    int n;
    scanf("%d", &n);

    for(int i = 0 ; i < n ; i++){
        int a1, a2;
        scanf("%d %d", &a1, &a2);
        location.push_back({a1, a2});
    }

    sort(location.begin(), location.end());

    for (auto const& entry : location)
    {
        std::cout << entry.first << ' ' << entry.second << '\n';
    }
}

 

여기서는 vector<pair<int, int>>를 활용하게 되었으니 머리에 잘 넣어야 겠다.