在C++中實現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 = {5, 2, 7, 1, 9, 3};
int max_index = argmax(vec);
std::cout << "The index of the maximum value is: " << max_index << std::endl;
return 0;
}
在上面的代碼中,首先定義了一個argmax函數,該函數接受一個整數類型的vector作為參數,然后使用std::max_element函數找到vector中的最大值的迭代器,并通過std::distance函數計算出最大值的索引位置。最后在main函數中調用argmax函數,找到并輸出最大值的索引位置。