# C++中实现argmin和argmax
python 中寻找一个数组中最大最小元素的下标注，可以使用numpy中的argmin和argmax函数：
a = np.array([40, 10, 100, 90, 20, 25])
print(np.argmin(a))
print(np.argmax(a))
# 1
# 2
C++中没有现成的函数可以用，可以通过如下方式自定义：
#include<algorithm>
template <typename ForwardIterator>
inline size_t argmin(const ForwardIterator begin, const ForwardIterator end)
{
return std::distance(begin, std::min_element(begin, end));
}
template <typename ForwardIterator>
inline size_t argmax(const ForwardIterator begin, const ForwardIterator end)
{
return std::distance(begin, std::max_element(begin, end));
}
int main()
{ size_t num = 6;
int values[num] = {40, 10, 100, 90, 20, 25};
size_t mi = argmin(values, values + num);
size_t ma = argmax(values, values + num);
std::cout << "argmin of values: " << mi << std::endl;
std::cout << "argmax of values: " << ma << std::endl;
return 0;
}
// argmin of values: 1
// argmax of values: 2
- 如上，
argmin和argmax函数只有一行，因此定义为了inline内联函数，避免函数跳转影响函数性能，inline函数通常定义在头文件中 argmin和argmax函数定义为了模板函数，ForwardIterator是C++中的前向迭代器，其结合了所有输入迭代器的功能和几乎所有输出迭代器的功能。因此，argmin和argmax函数的入参必须是ForwardIterator子类型，像指针，容器迭代器等。ForwardIterator类型的迭代器只能通过++前移，不能--后移动