从函数应用程序创建std :: vector

时间:2018-05-12 07:09:53

标签: c++ algorithm c++11 stl stdvector

给定一个对象的向量v1,以及对它们进行操作的函数,返回一个值,

std::vector<OBJ> v1; // already filled
bool foo(const OBJ&);

创建vector<bool> v2的最简洁方法是将foo应用于v1的每个元素?我宁愿不写循环,但我在STL中找不到合适的命令。

2 个答案:

答案 0 :(得分:2)

听起来像是std::transform的工作。

当然有一个循环,但它是transform的实现细节。

示例:

#include <vector>
#include <algorithm>

template<typename Obj, typename IsTrue>
std::vector<bool> truth(std::vector<Obj> const& v, IsTrue&& truth_func)
{
    auto result = std::vector<bool>(v.size());
    std::transform(std::begin(v), std::end(v), 
                   std::begin(result), 
                   std::forward<IsTrue>(truth_func));
    return result;
}


int main()
{
    std::vector<int> v { 1, 2, 3, 4, 5 };

    auto is_even = [](auto&& x)
    {
        return (x % 2) == 0;
    };

    auto vb = truth(v, is_even);
}

答案 1 :(得分:1)

我完全同意使用std::transform的解决方案。但是,我们可以使用std::for_each()中更简单的<algorithm>来完成此操作。

我们会考虑向量vec的每个元素(这是您的OBJ),insert boolfoo()返回到后端/末尾下一个向量v

如下所示。以下是实时操作:https://www.ideone.com/pyfqmF

#include <algorithm>
#include <iostream>
#include <vector>

using OBJ = int;

bool foo(const OBJ& a)
{ return a == 1; }

int main()
{
   std::vector< OBJ > vec = {1, 2, 3, 4};
   std::vector< bool > v;
   v.reserve(vec.size());

   std::for_each(vec.begin(), vec.end(), [&v](const OBJ& a)->void
      {
         v.insert(v.end(), foo(a) );
      });

   for(auto const& it: v) std::cout << it << std::endl;

   return 0;
}