可变计数函数

Variadic Count function

本文关键字：函数      更新时间：2023-10-16

我想写一个泛型函数，它计算容器中指定值的出现次数。

我希望调用看起来像这样:

struct Foo
{
    int GetI() { return i; }
    int i = 0;
};
Count(f, [](auto& f) { return f.GetI(); }, // from container 'f' with provided getter
            0, i0,    // count 0s into variable i0
            37, i37); // count 37s into variable i37

我已经为2个值编写了版本，现在我不确定，如何使其可变(特别是if elseif)。

template <typename Container, typename ValueGetter, typename SizeType, typename T1, typename T2>
void Count(Container& c, ValueGetter valueGetter, T1&& t1, SizeType& out1, T2&& t2, SizeType& out2)
{
    using GetVal = decltype(valueGetter(c[0]));
    static_assert(std::is_same<std::decay_t<T1>, GetVal>::value, "Types don't match!");
    static_assert(std::is_same<std::decay_t<T2>, GetVal>::value, "Types don't match!");
    for (auto& elm : c)
    {
        const auto& val = valueGetter(elm);
        if (val == t1) ++out1;
        else if (val == t2) ++out2;
    }
}

如何使它可变?

下面是ideone上使用

的小代码

您需要一个helper模板函数来展开可变参数包:

#include <iostream>
#include <vector>
template<typename value_type>
void Count_all(value_type &&value)
{
}
template<typename value_type, typename first_value, typename first_counter,
     typename ...Args>
void Count_all(value_type &&value, first_value &&value1,
           first_counter &&counter1,
           Args && ...args)
{
    if (value == value1)
        ++counter1;
    Count_all(value, std::forward<Args>(args)...);
}

template<typename Container, typename ValueGetter, typename ...Args>
void Count(const Container &c,
       ValueGetter &&getter,
       Args && ...args)
{
    for (const auto &v:c)
        Count_all(getter(v), std::forward<Args>(args)...);
}
int main()
{
    std::vector<int> i{1,2,3,3,5,9,8};
    int n_3=0;
    int n_8=0;
    Count(i, [](const int &i) { return i; },
          3, n_3,
          8, n_8);
    std::cout << n_3 << ' ' << n_8 << std::endl;
    return 0;
}
结果:

2 1

您可以接受对或元组参数并自己解压缩它们，而不是接受一堆纯可变参数:

template <typename Container, typename ValueGetter, typename... Pairs>
void Count(Container& c, ValueGetter valueGetter, Pairs&&... pairs)
{
    for (auto& elm : c)
    {
        auto check_value = [&](auto&& pair){
            const auto& val = valueGetter(elm);
            if (val == std::get<0>(pair)) ++std::get<1>(pair);
        };
        //Horrible hack to get around the lack of fold expressions
        (void)std::initializer_list<int> {
            (check_value(pairs), 0)...
        };
    }   
}

您可以添加任何static_assert，您想要提供更好的编译器错误误用。你可以这样使用:

Count(f, [](auto& f) { return f.GetI(); }, 
      std::forward_as_tuple(0, i0),
      std::forward_as_tuple(37, i37));

现场演示

相关文章：