温馨提示×

c++中argmax函数的高效写法

c++
小樊
96
2024-07-26 10:53:12
栏目: 编程语言
C++开发者专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在C++中,可以使用STL中的std::max_element函数来实现argmax(返回最大值的索引)的功能,示例如下:

#include <iostream>
#include <vector>
#include <algorithm>

int argmax(const std::vector<int>& vec) {
    return std::distance(vec.begin(), std::max_element(vec.begin(), vec.end()));
}

int main() {
    std::vector<int> vec = {4, 7, 2, 9, 5};

    int index = argmax(vec);
    std::cout << "Index of max element: " << index << std::endl;

    return 0;
}

这样可以在高效地找到vector中最大元素的索引。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:如何高效利用c++中的printf函数进行调试

0