PS/기타 알고리즘

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

프레딕 2022. 8. 1. 19:06
728x90

문제

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

입력

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

출력

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

코드

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

int N;
pair<int, int> tmp;
vector<pair<int, int>> vec;

bool compare(pair<int, int> a, pair<int, int> b) {
  if (a.second != b.second) {
    return a.second < b.second;
  }

  else {
    return a.first < b.first;
  }
}

int main() {
  cin >> N;

  for (int i = 0; i < N; ++i) {
    cin >> tmp.first >> tmp.second;
    vec.push_back(tmp);
  }

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

  for (int i = 0; i < N; ++i) {
    cout << vec[i].first << ' ' << vec[i].second << '\n';
  }

  return 0;
}

compare 함수 용도를 잘 알아두자.

728x90
반응형