C++中argmin和argmax的实现__c++ argmax

C++中argmin和argmax的实现

在Python中 argmin 和 argmax 这两个函数一般是用来就一列数中的最小值和最大值的索引。C++中我们如何实现呢？

实现思路

使用STL中的 std::min_element 函数求出最小值；
使用STL中的 std::distance 计算最小值跟迭代器的头部的距离；

实现代码

#include <algorithm>

template<class ForwardIterator>
inline size_t argmin(ForwardIterator first, ForwardIterator last)
{
    return std::distance(first, std::min_element(first, last));
}

template<class ForwardIterator>
inline size_t argmax(ForwardIterator first, ForwardIterator last)
{
    return std::distance(first, std::max_element(first, last));
}

测试代码

int main() {
    array<int, 7> numbers{2, 4, 8, 0, 6, -1, 3};
    size_t minIndex =  argmin(numbers.begin(), numbers.end());
    cout << minIndex << '\n';
    vector<float> prices = {12.5, 8.9, 100, 24.5, 30.0};
    size_t maxIndex = argmax(prices.begin(), prices.end());
    cout << maxIndex << '\n';
    return 0;
}

输出结果：

5
2

推荐文章

刚毅的莴苣 · CRLF和LF--团队开发难以排查的编码错误LF（Line Feed）代表“换行”，转义序列 \n，这个字符代表一行文本 - 掘金

4 月前

着急的冰淇淋 · java获取系统的网络信息 java如何抓取网络数据包_mob6454cc74c0fc的技术博客_51CTO博客

1 年前

欢快的开水瓶 · 服务器开启JMX监控 - clamaa - 博客园

1 年前

健壮的围巾 · Spark SQL/Hive 同一列的多行记录合并为一行_hive表中有同一个userid的两条记录,如何合并_光于前裕于后的博客-CSDN博客

1 年前

谦虚好学的铁链 · android - While converting X509CertificateHolder object into X509certificate am getting following exception - Stack Overflow

1 年前