문제
2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.
출력
첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.
내 접근
[[BOJ 11650, 좌표 정렬하기]] 의 개선점을 참고하여,
std::pair<int, int>를 사용해보고 싶었는데, 문제에서 제시하는 비교 방법이 달라서 못 쓸 것 같다.struct로Pos를 선언해서 좌표 값을 저장해야겠다.
구현
#include <iostream>
#define IO std::cin.tie(NULL), std::ios_base::sync_with_stdio(false)
#include <algorithm>
#include <vector>
struct Pos {
int x, y;
};
bool comp(const Pos &p1, const Pos &p2) {
if (p1.y == p2.y)
return p1.x < p2.x;
return p1.y < p2.y;
}
void solve() {
int n;
std::cin >> n;
std::vector<Pos> v(n);
for (int i = 0; i < n; i++) {
std::cin >> v[i].x >> v[i].y;
}
std::sort(v.begin(), v.end(), comp);
for (int i = 0; i < n; i++) {
std::cout << v[i].x << " " << v[i].y << '\n';
}
}
int main() {
IO;
solve();
return 0;
}개선점
for-loop 대신 range-based for 도 사용할 수 있다
루프 간에 인덱스 의미가 없는 경우 코드 표현이 더 명확해질 수 있다.
for (const auth& p : v) {
std::cout << p.x << ' ' << p.y << '\n';
}문제의 본질
이 문제에서 요구하는 것은
y 를 우선으로, 그 다음은 x 를 기준으로 정렬
Pos와 비교 함수가 필요하지 않을 수도 있음
std::pair 에 입력될 값 순서를 바꾼다면?
std::vector<std::pair<int, int>> v(n);
int x, y;
for (int i = 0; i < n; i++) {
std::cin >> x >> y;
v[i] = { y, x };
}- 비교 함수를 제거할 수 있음
- STL의 기본 정렬 규칙을 그대로 활용 가능
lambda 지역화
전역 비교 함수 대신 lambda 함수를 넘겨서, 비교 함수가 어떤 범위에서만 사용되는 건지 명시할 수 있다.
std::sort(v.begin(), v.end(),
[](const Pos& a, const Pos& b) {
return (a.y < b.y) || (a.y == b.y && a.x < b.x);
}
};