Question

我正在尝试找到一种优雅的方法来进行以下操作：

例如，我有以下课程：

class Foo {
    public: 
        SomeType bar;
}

foo.bar可以等于以下任何常量：BAR1，BAR2，BAR3，...，BARN

现在我有一个包含这样的元素的容器：std::vector<Foo> fooVec;，我想找到第一个foo.bar等于BAR1的元素，如果没有，那么我想要foo.bar等于BAR2的第一个元素，依此类推。

这是测试算法的示例代码：

#include <iostream>
#include <vector>

class Foo {
private:
    static int id_counter;
public:
    enum class Bar { _1, _2, _3, _4 };

    Bar bar;
    int id;

    Foo(Foo::Bar bar) : bar(bar), id(id_counter++) {}
};
int Foo::id_counter = 0;

const Foo& findFoo(const std::vector<Foo>& vec) {
    // Find first element verifying foo.bar == 1, or first verifying foo.bar == 2, etc...
}

int main() {
    std::vector<Foo> fooVec1({Foo(Foo::Bar::_2),
                              Foo(Foo::Bar::_4),
                              Foo(Foo::Bar::_1),
                              Foo(Foo::Bar::_3)});

    std::vector<Foo> fooVec2({Foo(Foo::Bar::_3),
                              Foo(Foo::Bar::_2),
                              Foo(Foo::Bar::_4),
                              Foo(Foo::Bar::_2)});

    std::cout << findFoo(fooVec1).id << std::endl; //Expected output: 2
    std::cout << findFoo(fooVec2).id << std::endl; //Expected output: 5

    return 0;
}

为了简化起见，我选择将SomeType替换为int，但假设使用关系运算符（<，<=，>，{{1} }）对这种类型没有意义。

所以，我想我在问问题时找到了解决方案，还有更好的方法吗？

>=

Answer 1

扩展Slava的评论：

const Foo& findFoo(const std::vector<Foo>& vec) {
    auto priority_less = [](const Foo & lhs, const Foo & rhs)
    {
        static std::unordered_map<Foo::Bar, int, EnumClassHash> priority {
            { Foo::Bar::_1, 1 },
            { Foo::Bar::_2, 2 },
            { Foo::Bar::_3, 3 },
            { Foo::Bar::_4, 4 },
        };

        return priority[lhs.bar] < priority[rhs.bar];
    };
    return *std::min_element(vec.begin(), vec.end(), priority_less);
}

注意：const Foo&的选择表示如果findFoo为空，则vec为UB。我会返回std::vector<Foo>::const_iterator或返回template <typename ForwardIterator> ForwardIterator findFoo(ForwardIterator, ForwardIterator)来匹配其他搜索算法

根据条件和优先级搜索容器中的项目

1 个答案: