如何从数组中的特定值创建向量?

时间:2015-06-17 09:30:41

标签: c++ arrays c++11 vector constructor

我将从代码开始:

#include <iostream>
#include <vector>
using namespace std;
struct A
{
    int color;

    A(int p_f) : field(p_f) {}
};
int main ()
{  
  A la[4] = {A(3),A(5),A(2),A(1)};
  std::vector<int> lv = {begin(la).color, end(la).color};//I would like to create vector from specific value from array la
  for (std::vector<int>::iterator it = fifth.begin(); it != fifth.end(); ++it) std::cout << ' ' << *it;
  return 0;
}

通常我想从数组中的特定值创建一个向量。 你可以看到la是一个数组,我想创建不包含整个la数组的向量,但只包含颜色。 vector(int)不是vector(A),哪个是vector {3,5,2,1},所以不是A,而只是int颜色。它也可以在C ++ 11中使用。感谢。

2 个答案:

答案 0 :(得分:10)

这应该有用。

std::vector<int> lv;
std::transform(std::begin(la), std::end(la), std::back_inserter(lv), [](const A& a){
    return a.color;
});

另外还有另一种方法:

重构您的结构以从方法中获取颜色:

struct A
{
    int color;

    A(int p_f) : color(p_f) {}

    int getColor() const {
        return color;
    }
};

在这种情况下,您可以使用bind

std::transform(std::begin(la), std::end(la), std::back_inserter(lv), std::bind(&A::getColor, std::placeholders::_1));

或者您也可以使用std::mem_fn方法缩短(感谢@Piotr S。):

std::transform(std::begin(la), std::end(la), std::back_inserter(lv), std::mem_fn(&A::getColor));

或者您可以将std::mem_fn用于数据成员。在这种情况下,您甚至不需要实现getter方法:

std::transform(std::begin(la), std::end(la), std::back_inserter(lv), std::mem_fn(&A::color));

答案 1 :(得分:1)

以下可能会有所帮助:

namespace detail
{
    using std::begin;
    using std::end;

    template <typename Container, typename F>
    auto RetrieveTransformation(const Container& c, F f)
        -> std::vector<std::decay_t<decltype(f(*begin(c)))>>
    {
        // if `F` return `const T&`, we want `std::vector<T>`,
        // so we remove reference and cv qualifier with `decay_t`.
        //
        // That handles additionally the case of lambda
        // The return type of lambda [](const std::string&s) { return s;}
        // - is `const std::string` for msvc
        // - is `std::string` for for gcc
        // (Note that the return type rules have changed:
        // see http://www.open-std.org/jtc1/sc22/wg21/docs/cwg_defects.html#1048)
        using F_Ret = std::decay_t<decltype(f(*begin(c)))>;
        std::vector<F_Ret> res;
        res.reserve(std::distance(begin(c), end(c)));
        for (const auto& e : c)
        {
            res.push_back(f(e));
        }
        return res;
    }

}

template <typename Container, typename F>
auto RetrieveTransformation(const Container& c, F f)
-> decltype(detail::RetrieveTransformation(c, f))
{
    return detail::RetrieveTransformation(c, f);
}

然后将其用作

std::vector<int> lv = RetrieveTransformation(la, std::mem_fn(&A::getColor));
// or
// auto lv = RetrieveTransformation(la, [](const A&a){return a.color;});

Live Demo