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

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

#include 
#include 
#include 

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 class="hljs-keyword">return 0;
}

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