백준 10814번 - 나이순 정렬 (p.s. 배울 게 많군)

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

10814번: 나이순 정렬

사고과정은 크게 두가지이다.

1. 변수가 두개이고 서로 연동되어 있으니, pair을 사용해볼 것인가

2. 아니면 그냥 클래스나 구조체로 묶어서 활용할 것인가

사실 풀면서 1번 생각이 났지만 pair을 사용해본 적이 없어서 문제는 클래스로 풀었지만 구글링해서 배운 결과를 공유해볼까한다.

1. pair을 활용

#include <iostream>
#include <vector>
#include <algorithm>
#include <utility>

using namespace std;

bool cmp (pair<int, string> u, pair<int, string> v)
{
    return u.first < v.first;
}

int main()
{
    int T;
    cin >> T;

    vector<pair<int, string>> vec(T);

    for (int i = 0; i < T; i++)
        cin >> vec[i].first >> vec[i].second;

    stable_sort(vec.begin(), vec.end(), cmp);

    for (int i = 0; i < T; i++)
        cout << vec[i].first << " " << vec[i].second << "\n";
}

코드가 매우 직관적이다.

sort 함수를 활용해 first, 즉 나이 순을 활용해서 정렬하고 있다.

2. 클래스를 활용

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Person
{
public:
	int age;
	string name;
	int index;
};
bool cmp(Person u, Person v)
{
	if (u.age < v.age)
		return true;
	else if (u.age == v.age)
		return u.index < v.index;
	else
		return false;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);

	int N;
	cin >> N;

	std::vector<Person> arr(N);
	for (int i = 0; i < N; i++)
	{
		cin >> arr[i].age >> arr[i].name;
		arr[i].index = i;
	}
		
	sort(arr.begin(), arr.end(), cmp);

	for (int i = 0; i < N; i++)
	{
		cout << arr[i].age << " " << arr[i].name << '\n';
	}

	return 0;
}

클래스를 활용할 경우 아까 1번의 pair 경우랑 야악간 다르다.

pair의 경우는 입력받은 값들이 말 그대로 순서대로 되어있지만

class의 경우는 만약 나이가 같을 경우 어떤 것을 우선으로 해야할지 처리해줘야한다.

그래서 class에 입력 순서를 받을 index라는 변수를 선언해서 순서를 저장해서 cmp 함수에 활용하였다.

 

개인적으로 의미있게 배운 문제이다. 복습 자주 해야징!