프로그래밍 공부/백준 (C++)
[C++/백준 11650번] 좌표 정렬하기
Rocketbabydolls
2023. 8. 7. 17:50
https://www.acmicpc.net/problem/11650
11650번: 좌표 정렬하기
첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.
www.acmicpc.net
해결 방법
정렬을 직접 구현해서 하려면 nlogn 시간복잡도를 가지는 정렬 알고리즘으로 정렬을 하면 될 듯 하다. 하지만 그렇게 하게 되면 조건이 생각보다 복잡해진다. stl vector 라이브러리를 사용하면 쉽게 풀 수 있다.
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<pair<int, int>> v;
int main() {
int N;
cin >> N;
int x, y;
for (int i = 0; i < N; i++) {
cin >> x >> y;
v.push_back({ x, y });
}
sort(v.begin(), v.end());
for (int i = 0; i < N; i++) {
cout << v[i].first << ' ' << v[i].second << '\n';
}
return 0;
}
이렇게 작성하면 Pair 의 첫 번째 인자를 기준으로 벡터 정렬을 한다.
두 번째 인자를 기준으로 정렬을 하기 위해선 함수를 정의해서 sort 의 세 번째 인자에 넣어줘야 한다고 한다. 아래 블로그를 참고했다.
알고리즘 - pair 원소를 가지는 vector의 정렬기준 변경하기
컴퓨터/IT/알고리즘 정리 블로그
chanhuiseok.github.io