如何设置std::binary_search中的指针类型

How to set needle type in std::binary_search

本文关键字：search 类型 指针 binary 设置 std 何设置      更新时间：2023-10-16

我有下面这个简单的程序，它将对一个项目进行binary_search:

#include <vector>
#include <algorithm>
#include <string>
#include <iostream>
class Record
{
public:
    Record() = default;
    Record(std::string name, int data) : mName(name), mData(data) { }
    std::string mName;
    int mData = 0;
};
int main(int, char**)
{
    std::vector<Record> recs;
    recs.emplace_back(Record("1", 1));
    recs.emplace_back(Record("55555", 2));
    recs.emplace_back(Record("333", 3));
    recs.emplace_back(Record("qwertyuiop", 4));
    recs.emplace_back(Record("22", 5));
    recs.emplace_back(Record("4444", 6));
    std::cout << "Unsorted:" << std::endl;
    for (auto& rec : recs)
    {
        std::cout << "Name: "  << rec.mName << " Data: " << rec.mData << std::endl;
    }
    std::cout << std::endl;
    std::stable_sort(recs.begin(), recs.end(), [](const Record& lhs, const Record& rhs) -> bool
    {
        return lhs.mName.length() < rhs.mName.length();
    });
    std::cout << "Sorted:" << std::endl;
    for (auto& rec : recs)
    {
        std::cout << "Name: " << rec.mName << " Data: " << rec.mData << std::endl;
    }
    std::cout << std::endl;
    if (std::binary_search(
        recs.begin(), 
        recs.end(),
        Record("qwertyuiop", 4),
        [](const Record& lhs, const Record& rhs) -> bool 
    {
        return lhs.mName < rhs.mName;
    }))
    {
        std::cout << "Found" << std::endl;
    }
    else
    {
        std::cout << "Not found" << std::endl;
    }

    return 0;
}

如何根据另一种类型的T搜索记录向量?例如std::string而不是Record?

类似:

 if (std::binary_search(
            recs.begin(), 
            recs.end(),
            "qwertyuiop",
            [](const Record& lhs, const std::string& rhs) -> bool 
        {
            return lhs.mName < rhs.mName;
        }))
        {
            std::cout << "Found" << std::endl;
        }
        else
        {
            std::cout << "Not found" << std::endl;
        }

是我想要的-基本上我不想构造一个记录来搜索它，因为这对我来说是一个性能问题。

您可以创建一个处理异构比较的函数对象类型:

struct comp_t
{
    bool operator()(Record const& lhs, std::string const& rhs) const {
        return lhs.mName < rhs;
    }
    bool operator()(std::string const& lhs, Record const& rhs) const {
        return lhs < rhs.mName;
    }
};

那么你可以这样调用binary_search:

std::binary_search(recs.begin(), recs.end(),
                   std::string("qwertyuiop"),
                   comp_t{})

您可以通过添加一个implicit构造函数来将std::string转换为Record，并添加一个操作符来将Record转换为std::string。

operator std::string() const {
    return mName;
}
Record(std::string name) : mName(name), mData(int()) {}

或者您可以将现有的构造函数转换为

Record(std::string name, int data = int()) : mName(name), mData(data) {}

但是无论如何，你需要创建一个std::string对象来搜索。

/// create std::string object from "qwertyuiop"
if (std::binary_search(recs.begin(), recs.end(), std::string("qwertyuiop"), 
    [](const Record & lhs, const std::string & rhs) -> bool 
        {
            return lhs.mName < rhs;    /// rhs is a std::string type
        })) {
    std::cout << "Found" << std::endl;
}
else {
    std::cout << "Not found" << std::endl;
}

相关文章：