c++ - 向量的 ArgMin在 C++ 中？

Question

我想在 C++ 中找到最小值的索引std::vector<double>。这是一个有点冗长的实现：

//find index of smallest value in the vector
int argMin(std::vector<double> vec)
{
    std::vector<double>::iterator mins = std::min_element(vec.begin(), vec.end()); //returns all mins
    double min = mins[0]; //select the zeroth min if multiple mins exist
    for(int i=0; i < vec.size(); i++)
    {
        //Note: could use fabs( (min - vec[i]) < 0.01) if worried about floating-point precision
        if(vec[i] == min)    
            return i;
    }
    return -1;
}

（如果您发现上述实现中有任何错误，请告诉我。我对其进行了测试，但我的测试并不详尽。）

我认为上述实现可能是轮子改造；如果可能的话，我想使用内置代码。为此，是否有对 STL 函数的单行调用？或者，有人可以建议一个更简洁的实现吗？

Answer 1

您可以使用标准min_element功能：

std::min_element( vec.begin(), vec.end() );

它返回一个迭代器，指向迭代器范围内的最小元素。由于您想要一个索引并且您正在使用vectors，因此您可以从中减去生成的迭代器vec.begin()以获得这样的索引。

如果您需要自定义比较，则函数或函数对象还有一个额外的重载。