c++ - 在2D vector 上使用STL算法提取第一列

Question

我正在尝试学习使用更多的STL算法。我无法查看是否存在更标准的编写以下代码的方式。

std::vector<std::vector<std::string>> myVec;
std::vector<std::string> myInnerVec;
for(int y=0;y < myVec.size();y++){
    myInnerVec.emplace_back(myVec[y][0]);
}

在这里，我从 vector myVec中提取第一列，并将其放入myInnerVec中。可以使用任何STL魔术来改进此代码吗？

Answer 1

例如，您可以使用标准算法std::transform。

这是一个演示程序。

#include <iostream>
#include <string>
#include<vector>
#include <iterator>
#include <algorithm>

int main() 
{
    std::vector<std::vector<std::string>> v =
    {
        { "A", "B", "C" },
        { "D", "E", "F" }
    };

    std::vector<std::vector<std::string>>::size_type column;

    auto l = [&column = column]( const auto &row )
    {
        return row[column];
    };

    column = 0;

    std::vector<std::string> v1;

    v1.reserve( v.size() );

    std::transform( std::begin( v ), std::end( v ), std::back_inserter( v1 ), l );

    for ( const auto &item : v1 ) std::cout << item << ' ';
    std::cout << '\n';

    column = 1;

    std::transform( std::begin( v ), std::end( v ), std::begin( v1 ), l );

    for ( const auto &item : v1 ) std::cout << item << ' ';
    std::cout << '\n';

    return 0;
}

它的输出是

A D 
B E