[baekjoon] 10814 : 나이순 정렬

문제

---

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

 

입력

---

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

 

출력

---

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

 

예제 입출력

---

 

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

10814번: 나이순 정렬

 

문제 풀이

---

vector의 pair함수를 사용해 넣어주고 first만 정렬을 해야하는데 그냥 sort만 사용하여 벡터를 정렬하면 뒤에있는 이름도 함께 정렬이 되어 틀리게 된다. 따라서 sort안에 조건을 추가로 넣어 이름은 정렬되지 않고 입력받은 순서로 정렬되게 해줘야 한다. sort 괄호 안에 begin과 end를 사용하고 그뒤에 조건을 달아주면 되는데 람다함수를 사용하여 첫번째 값을 기준으로 오름차순 정렬해주고 같을 경우 순서를 유지시켜주도록 만들면 된다.

그냥 sort로 넣으면 visual studio에서는 문제없이 잘 돌아가는데 백준에서는 무슨 문제인지 틀렸다고 나와 알아보니 sort대신 stable_sort를 사용후 sort와 동일하게 조건을 적으면 바로 풀리게 된다.

 

최종 코드

---

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

int main() {
	int N;
	vector<pair<int, string>> p;
	int a;
	string n;
	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> a >> n;
		p.push_back(make_pair(a, n));
	}
	stable_sort(p.begin(), p.end(), [](const pair<int, string>& a, const pair<int, string>& b) {
		if (a.first != b.first)
			return a.first < b.first;
		return false; // return false로 함으로써 두 pair의 순서를 변경하지 않도록 함.
	});
	for (int i = 0; i < N; i++) {
		cout << p[i].first << " " << p[i].second << "\n";
	}
	
}