温馨提示×

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

c++
小樊
85
2024-07-26 10:53:12
栏目: 编程语言

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

0