std::vector 에서 argmax, argmin 값을 계산하자.

데이터를 다루다 보면 min, max 값 뿐만 아니라 해당 값의 index, 즉, argmin-max 가 궁금할 때가 있다

파이썬에서는 argmin, argmax 함수가 있어 계산하기 간단하지만 C++에서는 해당 함수가 없다

 

대신 사용할 수 있는 방법이 있어서 기록

---

vector<int> v;
v.push_back(1);
v.push_back(3);
v.push_back(9);
v.push_back(0);
v.push_back(5);


// max
int nMax = *max_element(v.begin(), b.end());
// 9

// min 
int nMin = *min_element(v.begin(), b.end());
// 0

// argmax
int nArgMax = max_element(v.begin(), b.end()) - v.begin();
// 2

// argmin 
int nArgMin = min_element(v.begin(), b.end()) - v.begin();
// 3