前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >C++中argmin和argmax的实现

C++中argmin和argmax的实现

作者头像
卡尔曼和玻尔兹曼谁曼
发布2019-01-22 09:30:59
2K0
发布2019-01-22 09:30:59
举报
文章被收录于专栏:给永远比拿愉快

C++中argmin和argmax的实现

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

实现思路

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

实现代码

代码语言:javascript
复制
#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));
}

测试代码

代码语言:javascript
复制
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;
}

输出结果:

代码语言:javascript
复制
5
2
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018年10月13日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • C++中argmin和argmax的实现
    • 实现思路
      • 实现代码
        • 测试代码
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档