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

문제

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

 

입력

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

 

출력

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

 

예제 입력 1 

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

 

예제 출력 1 

1 -1
1 1
2 2
3 3
3 4

 

알고리즘 분류

• 정렬

 

문제 출처

https://www.acmicpc.net/problem/11650

11650번: 좌표 정렬하기

 

문제 해결 방법

• 벡터와 sort() 함수를 이용하여 문제를 해결하였다.
• sort() 함수의 세 번째 인자에 넣어 콜백할 함수를 다음과 같이 정의하였다.

bool compare(pair<int, int> a, pair<int, int> b) {
    if (a.first == b.first) {    // x 좌표가 같으면
        return a.second < b.second;    // y 좌표를 오름차순으로 정렬
    }
    return a.first < b.first;    // x 좌표를 오름차순으로 정렬
}

• 벡터(xy)의 인자로 pair를 이용하여 int 형 변수 2개를 입력 받고, 첫 번째 변수(xy.first)의 크기가 같을 경우 두 번째 변수(xy.second)의 크기를 비교하여 오름차순으로 정렬하도록 하였다. 그렇지 않을 경우 첫 번째 변수를 기준으로 오름차순으로 정렬하도록 하였다.
• pair를 이용하여 2개의 인자를 입력 받을 때는 중괄호({ })를 이용하여 다음과 같이 입력 받는다.

xy.push_back({x, y});

 

코드

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

int N, x, y;
vector<pair<int, int>> xy;

bool compare(pair<int, int> a, pair<int, int> b) {
    if (a.first == b.first) {    // x 좌표가 같으면
        return a.second < b.second;    // y 좌표를 오름차순으로 정렬
    }
    return a.first < b.first;    // x 좌표를 오름차순으로 정렬
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> N;

    for (int i = 0; i < N; i++) {
        cin >> x >> y;
        xy.push_back({x, y});
    }

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

    for (auto i : xy) {
        cout << i.first << " " << i.second << '\n';
    }

    return 0;
}

 

채점 결과

 

참고

• [단계별로 풀어보기] > [정렬]
• 실버V