C++ lower_bound() 搜索最接近目标值的元素

Question

假设我有一个vector，它的元素是int 类型。如何优雅地使用std::lower_bound() 来寻找最接近目标值的元素？

我写的代码如下图：

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

int main() {
    vector<int> vec {3,4,5,10};
    int target = 6;

    auto found {
        lower_bound(
            vec.begin(),
            vec.end(),
            target,
            [=] (int ele1, int ele2) {
                return (abs(ele1 - target) > abs(ele2 - target));
            }
        )
    };

    cout << (found-vec.begin()) << endl;

    return 0;
}

它返回4的索引，这意味着found = vec.end()。相反，我想获得2 的索引。我哪里搞砸了？

Answer 1

您在lower_bound 中使用的谓词与用于vec 排序的谓词不同。所以你的代码 sn-p 有未定义的行为。

您仍然可以使用lower_bound 来简单地查找target。

auto attempt = lower_bound(vec.begin(), vec.end(), target);

现在您实际查找的迭代器将是返回的迭代器，或者是下一个元素的迭代器：

auto found = attempt == vec.end() ? 
             vec.end() - 1 :   // last element is closest
             attempt == vec.begin() ? 
             vec.begin() :  // first element is closest
             abs(*attempt - target) < abs(*(attempt - 1) - target) ? 
             // somewhere in the middle, then the previous element could be closer
             attempt : attempt - 1;   

这假定 vec 不为空，因此您可能还需要进行该检查。

这是demo。