(七)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类型的迭代器只能通过++前移，不能--后移动

欢迎访问个人网络日志🌹🌹知行空间🌹🌹

参考资料

• 1.https://www.javatpoint.com/cpp-forward-iterator