如何在C++中实现argmax功能

avatar
作者
筋斗云
阅读量:0

在C++中实现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 = {5, 2, 7, 1, 9, 3};     int max_index = argmax(vec);          std::cout << "The index of the maximum value is: " << max_index << std::endl;      return 0; } 

在上面的代码中,首先定义了一个argmax函数,该函数接受一个整数类型的vector作为参数,然后使用std::max_element函数找到vector中的最大值的迭代器,并通过std::distance函数计算出最大值的索引位置。最后在main函数中调用argmax函数,找到并输出最大值的索引位置。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!