본문 바로가기

공부/C++

[C/C++] 좌표 정렬하기 2 (11651)

문제

 

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

 

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

 

 

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

 

입력

 

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

 

출력

 

1 -1
1 2
2 2
3 3
0 4

 

풀이

 

분명 아까와 다른 문제라는 것을 미리 명시하고 시작하도록 하겠다. 그래도 거의 비슷한 문제다. 정말 한 글자만 바꾸니 바로 해결이 가능하긴하다. 일단 이번에는 아까와는 다르게 y좌표를 기준으로 정렬이 되었다.

 

하지만 방식은 y좌표 먼저 정렬 그 후 x좌표 정렬을 진행하면 되기 때문에 두 수를 바꿔주기만 하면 되는 문제이다.

 

코드

 

#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", &a2, &a1);
        location.push_back({a1, a2});
    }

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

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

 

이전 포스팅에서 적지 않은 녀석에 대해서 알아두고 가려고 한다. 바로 auto const&라는 녀석인데 모양을 고민하게 되는 순간에 var 처럼 쓸 수 있는 듯 하다. for(auto const& ~~~ : ~~~)로 기억하면 좋을 것 같다.