Question

我有一个指向A类的指针向量，我希望使用STL按int键进行排序。为此，我在A类中定义了operator <

bool operator< (const A &lhs, const A &rhs){
    return (lhs.key < rhs.key);
};

在我的插入函数中它看起来像

vector<A*>::iterator it = lower_bound(vec.begin(), vec.end(), element);
vec.insert(it, element);

我希望lower_bound返回可以放置新元素的第一个位置，但它不起作用。插入带有键0,1,2,3的对象将导致向量的顺序不正确（2,3,1,0）。那是为什么？

也许我也可以在这个对象上使用比较器：

compare function for upper_bound / lower_bound

但我的代码出了什么问题？

Answer 1

从您的示例中，您使用的是指针向量：std::vector<A*>。因此，您需要为传递到std::lower_bound的指针定义比较：

bool less_A_pointer (const A* lhs, const A* rhs)
{
    return (lhs->key < rhs->key);
}

auto it = std::lower_bound(vec.begin(), vec.end(), element, less_A_pointer);
//...

当然，问题是为什么你首先使用指针 - 只是存储A对象，除非你真的需要。如果您确实需要存储指针，请查看std::shared_ptr，它将为您处理：

std::vector<std::shared_ptr<A>> v;
std::shared_ptr<A> element(new A(...));
auto it = std::lower_bound(v.begin(), v.end(), element); //Will work properly

您也可以使用lambda而不必编写自由函数：

auto it = std::lower_bound(v.begin(), v.end(), 
                          [](const A* lhs, const A* rhs)
                          { return lhs->key < rhs->key; });

Answer 2

如果您真的需要std::vector指针，可以考虑使用智能指针，例如std::shared_ptr。
原始指针如果观察指针就可以了，但通常你应该不使用 raw 拥有< / strong>指针（除非在某些特殊情况下）。

您可以将lambda传递给std::lower_bound()，以指定排序条件（在这种情况下，比较关键数据成员）。

此外，您可以使用C ++ 11的std::vector<std::shared_ptr<A>>::iterator关键字，而不是明确地为std::lower_bound()的返回值编写auto，这使得代码在这种情况下更具可读性。

随后是一个可编译的代码示例（使用g ++ 4.8.0编译）：

#include <algorithm> // for std::lower_bound #include <iostream> // for console output #include <memory> // for std::make_shared, std::shared_ptr #include <string> // for std::string #include <vector> // for std::vector using namespace std; // Test data structure struct A { int Key; string Data; A() : Key(0) {} A(int key, const string& data) : Key(key), Data(data) {} }; ostream& operator<<(ostream& os, const A& a) { os << "(key=" << a.Key << ", data=\"" << a.Data << "\")"; return os; } void Print(const vector<shared_ptr<A>> & v) { cout << "[ "; for (const auto & p : v) { cout << *p << " "; } cout << " ]\n"; } int main() { // Test container vector<shared_ptr<A>> v; // Test data const char* data[] = { "hello", "world", "hi", nullptr }; // Index in data array int i = 0; // Insertion loop while (data[i] != nullptr) { // Create new element on the heap auto elem = make_shared<A>(i, data[i]); // Find ordered insertion position auto it = lower_bound(v.begin(), v.end(), elem, [](const shared_ptr<A>& lhs, const shared_ptr<A>& rhs) { return lhs->Key < lhs->Key; } ); // Insert in vector v.insert(it, elem); // Move to next data i++; } // Show the result Print(v); }

这是输出：

[ (key=2, data="hi") (key=1, data="world") (key=0, data="hello") ]

如何使用lower_bound将值插入到排序向量中

2 个答案: