# C++中实现argmin和argmax

python 中寻找一个数组中最大最小元素的下标注,可以使用numpy中的argminargmax函数:

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
  • 如上,argminargmax函数只有一行,因此定义为了inline内联函数,避免函数跳转影响函数性能,inline函数通常定义在头文件中
  • argminargmax函数定义为了模板函数,ForwardIteratorC++中的前向迭代器,其结合了所有输入迭代器的功能和几乎所有输出迭代器的功能。因此,argminargmax函数的入参必须是ForwardIterator子类型,像指针,容器迭代器等。ForwardIterator类型的迭代器只能通过++前移,不能--后移动
(adsbygoogle = window.adsbygoogle || []).push({});

# 参考资料