[Baekjoon] 11650 : 좌표 정렬하기 (C++)

Baekjoon Online Judge 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

---

코드

정렬 알고리즘을 직접 해보려고 했는데, 몇 번의 시도에도 시간초과가 나와서 나중에 정렬 알고리즘을 다시 한 번 공부해야 겠다는 생각을 하게 되었다.

#include <iostream>
#include <algorithm> // sort()

using namespace std;

typedef struct {  // 좌표 구조체
  int x, y;
} Point;

bool cmp(const Point &p1, const Point &p2) {
  // 좌표의 x값 오름차순, 같을 시 y값 오름차순
  if(p1.x < p2.x)
    return true;
  else if(p1.x == p2.x)
    return p1.y < p2.y;
  else
    return false;
}

int main(void) {
  int N; // 입력할 좌표의 수
  Point* points; // 좌표를 저장할 공간

  cin >> N;
  points = new Point[N]; // 입력할 수 만큼 동적할당
  for(int i = 0; i < N; i++) // 좌표 입력
    cin >> (points[i]).x >> (points[i]).y; 

  sort(points, points+N, cmp); // 오름차순 정렬

  for(int i = 0; i < N; i++) // 정렬한 좌표 출력
    cout << points[i].x << " " << points[i].y << '\n';

  delete[] points; // 동적할당 해제
  return 0;
}