11650 번 : 좌표 정렬하기

문제는 다음과 같습니다.

https://www.acmicpc.net/problem/11650

11650번: 좌표 정렬하기

 

풀이 코드는 다음과 같습니다.

 

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

bool cmp(pair<int,int>p1,pair<int,int>p2)
{
	if(p1.first==p2.first)
		return p1.second<p2.second;
	else
		return p1.first<p2.first;
}
int main()
{
	int n;
	cin>>n;
	vector<pair<int,int>>v;
	for(int i{0};i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		v.push_back(make_pair(x,y));
		
	}
	sort(v.begin(),v.end(),cmp);
	for(int i{0};i<n;i++)
	{
		cout<<v[i].first<<" "<<v[i].second<<"\n";
	}
	return 0;
}

정렬은 3번째 요소에 사용자 정의 함수나 오름차순, 내림차순을 정할 수 있는 인자를 추가할 수 있습니다.

이번 문제의 경우에는 두 개의 수를 하나로 묶어서 첫 번째 원소로 정렬을 하고 만약 같다면 두 번째 원소로 정렬하도록 하라고 했으므로 pair를 통해 두 수를 하나로 묶고 조건에 따라 함수를 만들어주어 sort에 넣으면 풀 수 있습니다.