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

在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中最大元素的索引。