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