[c++/벡터] vector에서 pair 사용하기

벡터 입력을 받다 보면 int형 변수와 string형 변수 두가지 다 받아서 정렬을 해야 되는 경우가 있다.

이런 상황에서 사용할 수 있는 것이 pair 함수이다.

pair를 사용하기 위해서는 헤더에 #include <utility> 헤더를 추가한다.

​

​

vector < pair <type, type> > 변수 ;

- 2개의 각각 지정한 타입의 값을 저장

​

- 저장한 값은 .first 와 .second 로 각각 접근 가능

-> .first는 첫번째 인자 반환, .second는 두번째 인자 반환

​

- make_pair(value1, value2)

​

<first 기준 정렬>

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

int main() {
	vector < pair<int, string>> v;

	v.push_back(make_pair(10, "ㄱ"));
	v.push_back(make_pair(50, "ㄴ"));
	v.push_back(make_pair(30, "ㄷ"));
	v.push_back(make_pair(20, "ㄹ"));
	v.push_back(make_pair(40, "ㅁ"));

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

	cout << v[0].first << " " << v[0].second << endl;
	cout << v[1].first << " " << v[1].second << endl;
	cout << v[2].first << " " << v[2].second << endl;
	cout << v[3].first << " " << v[3].second << endl;
	cout << v[4].first << " " << v[4].second << endl;
}

<second 기준 정렬>

내림차순 정렬은 compare 함수에서 부등호 방향 바꿔주면 된다.

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


typedef pair<int, int> v;

bool compare(v x, v y) {
	return x.second < y.second;
}

int main() {

	vector<v> v;
	v.push_back(make_pair(10, 20));
	v.push_back(make_pair(50, 10));
	v.push_back(make_pair(30, 60));
	v.push_back(make_pair(20, 50));
	v.push_back(make_pair(60, 30));

	sort(v.begin(), v.end(), compare);

	cout << v[0].first << " " << v[0].second << endl;
	cout << v[1].first << " " << v[1].second << endl;
	cout << v[2].first << " " << v[2].second << endl;
	cout << v[3].first << " " << v[3].second << endl;
	cout << v[4].first << " " << v[4].second << endl;
}