10. C++ STL sort() 함수 다루기(2)

pair와 vector 라이브러리 사용해서 정렬하는 방법이다.

 

pair는 한 쌍의 데이터를 사용할 수 있는 라이브러리이다.

아무런 조건 없이 단순 정렬을 하는 경우 처음에 있는 항목 기준으로 정렬된다. (ex 하단 코드에서는 int 기준으로 정렬된다.)

 

문제 1. 

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

using namespace std;

int main(void) {
	vector<pair<int, string> > v;
    //v.push_back : 리스트 마지막에 삽입
	v.push_back(pair<int, string>(90, "박한울"));
	v.push_back(pair<int, string>(85, "이태일"));
	v.push_back(pair<int, string>(82, "나동빈"));
	v.push_back(pair<int, string>(98, "강종구"));
	v.push_back(pair<int, string>(79, "이상욱"));
	
    //처음부터 끝까지 정렬
	sort(v.begin(), v.end());
    
    //v.size : vector의 크기
    //v[i].second : 두번째 값 즉 하단 코드에서는 string값을 뜻함.
	for(int i = 0; i < v.size(); i++) {
		cout << v[i].second << ' ';
	}
	return 0;
}

 

출력 결과 : int기준 오름차순으로 string이 출력되었다.

 

 

문제 2.  학생을 나타낼 수 있는 정보가 이름, 성적, 생년월일일 때 학생을 성적 순서대로 나열하고자 한다. 다만 성적이 동일한 경우 나이가 더 어린 학생이 더 우선순위가 높다.

 

학생 명단

학생 1: 나동빈/90점/1996-12-22
학생 2: 이태일/97점/1993-05-18
학생 3: 박한울/95점/1993-02-03
학생 4: 이상욱/90점/1992-12/07
학생 5: 강종구/88점/1990-03-02

 

pair에 pair가 들어가는 방식.

sort안에 compare을 넣고 compare(정렬 기준)을 앞에서 정의해줌.

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

using namespace std;

bool compare(pair<string, pair<int, int> > a,
		pair<string, pair<int, int> > b) {
	if(a.second.first == b.second.first) {
		return a.second.second > b.second.second;
	} else {
		return a.second.first > b.second.first;
	}
} 

int main(void) {
	vector<pair<string, pair<int, int> > > v;
	v.push_back(pair<string, pair<int, int> >("나동빈", make_pair(90, 19961222)));
	v.push_back(pair<string, pair<int, int> >("이태일", make_pair(97, 19930518)));
	v.push_back(pair<string, pair<int, int> >("박한울", make_pair(95, 19930203)));
	v.push_back(pair<string, pair<int, int> >("이상욱", make_pair(90, 19921207)));
	v.push_back(pair<string, pair<int, int> >("강종구", make_pair(88, 19900302)));
	
	sort(v.begin(), v.end(), compare);
	for(int i = 0; i < v.size(); i++) {
		cout << v[i].first << ' ';
	}
	return 0;
}

 

출력 결과

 

만약 변수의 개수가 3개가 넘어가는 경우에는 지난 글 C++ STL sort() 함수 다루기(1)에서 처럼 클래스를 정의해서 정렬하는 것이 더욱 효율적이다.