从类成员生成容器

时间:2018-01-19 10:32:54

标签: c++ algorithm oop stl stl-algorithm

我想知道std库中是否有一个算法能够将包含类类型的容器转换为另一个容纳该类类型成员的容器。

所以我基本上寻找的是下面代码中像copy_acces这样的函数的STL替代方案:

template<class InputIterator, class OutputIterator, class Accessor>
OutputIterator copy_acces(InputIterator first, InputIterator last,
                          OutputIterator result, Accessor acc) 
{
    while (first!=last) {
        *result = acc(*first);
        ++result;
        ++first;
    }

    return result;
}

可以这样使用:

using namespace std;

struct A {
    int i;    
};

int main()
{
    vector<A> vecA = {{0}, {1}, {2}};
    vector<int> vecInt;

    copy_acces(vecA.begin(), vecA.end(), 
               back_insert_iterator< vector<int> >(vecInt), 
               [](A a){ return a.i; });

    for(int i : vecInt) {
        cout << i << endl;   
    }
}

2 个答案:

答案 0 :(得分:3)

对于像int那样廉价的默认构造,我会像这样使用std::transform

std::vector<int> vecInt(vecA.size());
std::transform(vecA.begin(), vecA.end(), vecInt.begin(), [](const A& a){ return a.i; });

Live demo

对于非默认构造或默认构造昂贵的内容,您无法或者不想首先构建正确大小的向量。您仍然可以将std::transformback_inserter一起使用,但我开始认为一个简单的基于范围的for循环更容易理解:

std::vector<ExpensiveToConstruct> vecB;
vecB.reserve(vecA.size());

for (const auto& a : vecA) {
    vecB.push_back(a.e);
}

Live demo

答案 1 :(得分:3)

尝试std::transform,例如:

#include <iostream>     // std::cout
#include <algorithm>    // std::transform
#include <vector>       // std::vector

using namespace std;

struct A {
    int i;    
};

int main()
{
    vector<A> vecA = {{0}, {1}, {2}};
    vector<int> vecInt;

    vecInt.resize(vecA.size());
    std::transform (vecA.begin(), vecA.end(), vecInt.begin(), [](A a){ return a.i; });

    for(int i : vecInt) {
        cout << i << endl;   
    }
}

输出:

0
1
2