溫馨提示×

c++ argmax的最佳實踐案例

c++
小樊
87
2024-07-26 10:54:10
欄目: 編程語言

在C++中，argmax通常指的是找到數(shù)組中最大值的索引。以下是argmax的一個最佳實踐案例：

#include <iostream>
#include <vector>
#include <algorithm>

int argmax(const std::vector<int>& arr) {
    auto max_element = std::max_element(arr.begin(), arr.end());
    return std::distance(arr.begin(), max_element);
}

int main() {
    std::vector<int> arr = {10, 5, 20, 15, 30};
    
    int max_index = argmax(arr);
    
    std::cout << "The index of the maximum element is: " << max_index << std::endl;
    
    return 0;
}

在這個例子中，我們首先定義了一個argmax函數(shù)，它接受一個整數(shù)vector作為參數(shù)，并返回該vector中最大元素的索引。我們使用std::max_element函數(shù)來找到最大元素的迭代器，然后使用std::distance函數(shù)獲取該迭代器在vector中的索引。

在main函數(shù)中，我們創(chuàng)建了一個包含一些整數(shù)的vector，并調(diào)用argmax函數(shù)來找到最大元素的索引，最后將結(jié)果打印出來。

這個實例展示了如何在C++中使用STL算法來實現(xiàn)argmax的功能。

0 踩