在`std::upper_bound` 的自定义比较器函数中获取当前元素的索引

Question

我正在尝试使用std::upper_bound来查找std::vector<double> xpositions元素的std::vector<double> xpositions 。 xpositions中每个元素的索引需要用于索引多维数组。 我试过了

upper_bound(xpositions.cbegin(), xpositions.cend(), value,
            [](const double& element, const double& value){
                // get index of current element
                const auto index = std::distance(xpositions.cbegin(), &element);
                // look up multidimensional array
                bigarray[index];
            }));

但这不会编译，因为&element无法转换为迭代器。 有没有办法在不执行潜在昂贵的std::find情况下获取element的索引？

Answer 1

vector 中的元素存储在连续区域中，简单的指针算法将完成这项工作：

const auto index = &element - &xpositions[0];

您还需要在 lambda 中通过引用捕获xpositions 。

如果你想使用distance你必须将向量的非常量迭代器传递给upper_bound ，并且谓词应该对双upper_bound进行非常量引用：

upper_bound(xpositions.begin(), xpositions.end(), value,
            [&](const double& value, double& element){
                // get index of current element
                auto index = std::distance(&xpositions[0],&element);