如何從數組的特定值創建向量？

Question

我將從碼開始：

#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）不是矢量（A），哪個矢量{3,5,2,1}，所以不是A，而只是int的顏色。它也可以在C++ 11中使用。謝謝。

Answer 1

這應該工作。

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)); 

Answer 2

以下可能會有所幫助：

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