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