5장 정렬 - (3) 11650번 좌표 정렬하기(비교 함수, 연산자 오버로딩 사용)

1. 비교 함수 사용

#include <iostream>

#include <vector>

#include <algorithm>

using namespace std;

struct Point {

int x, y;

};

bool cmp(const Point& u, const Point& v) {

if (u.x < v.x)

return true;

else if (u.x == v.x)

return u.y < v.y;

else

return false;

}

int main() {

int n;

cin >> n;

vector<Point> a(n);

for (int i = 0; i < n; i++) {

cin >> a[i].x;

cin >> a[i].y;

}

sort(a.begin(), a.end(), cmp);

for (int i = 0; i < n; i++) {

cout << a[i].x << " " << a[i].y << '\n';

}

return 0;

}

2. 연산자 오버로딩 사용

#include <iostream>

#include <algorithm>

#include <vector>

using namespace std;

struct Point {

int x, y;

bool operator <(const Point& v) const {

if (x < v.x)

return true;

else if (x == v.x) 

return y < v.y;

else

return false;

}

};

int main(void) {

int n;

cin >> n;

vector<Point> a(n);

for (int i = 0; i < n; i++) {

cin >> a[i].x;

cin >> a[i].y;

}

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

for (int i = 0; i < n; i++) {

cout << a[i].x << " " << a[i].y << '\n';

}

return 0;

}