溫馨提示×

c++中argmax函數的高效寫法

c++
小樊
111
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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女