在C ++中堆叠/连接2D向量

时间:2015-08-13 16:28:05

标签: c++ vector stack 2d concatenation

我正在尝试堆叠/垂直连接2D矢量。对于1D向量,我有这样的东西:

#include<iostream>
#include<vector>

using namespace std;

int main()
{
    vector< vector<int> > res;//(2,vector<int>(3,0.0));

    vector<int>a = {1,1,1};
    vector<int>b = {6,6,6};

    res.push_back(a);
    res.push_back(b);

    for(int i = 0; i < res.size(); i++)
    {

        for(int j = 0; j < res[0].size(); j++)
        {
            cout << res[i][j] << ",";
        }

        cout << endl;
    }

    return 0;
}

所以得到的2D矢量(矩阵):

1, 1, 1,
6, 6, 6,

是向量a和b的堆叠/垂直连接版本。现在,我的a和b是2D向量而不是1D向量:

 vector< vector<int> >a = {{1,2,3},
                            {2,2,2}};

  vector< vector<int> >b = {{4,5,6},
                            {6,6,6}};

我如何将它们堆叠成一个尺寸为4 x 3的结果矩阵:

1, 2, 3,
2, 2, 2,
4, 5, 6,
6, 6, 6,

因为,简单的push_back()不会这样做。

1 个答案:

答案 0 :(得分:0)

你的意思是以下几点吗?

#include <iostream>
#include <vector>

int main()
{
    std::vector<std::vector<int>> a = { { 1, 2, 3 }, { 2, 2, 2 } };
    std::vector<std::vector<int>> b = { { 4, 5, 6 }, { 6, 6, 6 } };
    std::vector<std::vector<int>> res;

    res = a;
    res.insert( res.end(), b.begin(), b.end() );

    for ( const auto &row : res )
    {
        for ( int x : row ) std::cout << x << ' ';
        std::cout << std::endl;
    }        
}    

程序输出

1 2 3 
2 2 2 
4 5 6 
6 6 6 

您也可以使用push_back。例如

#include <iostream>
#include <vector>
#include <functional>

int main()
{
    std::vector<std::vector<int>> a = { { 1, 2, 3 }, { 2, 2, 2 } };
    std::vector<std::vector<int>> b = { { 4, 5, 6 }, { 6, 6, 6 } };
    std::vector<std::vector<int>> res;
    res.reserve( a.size() + b.size() );

    for ( auto &r : { std::cref( a ), std::cref( b ) } )
    {
        for ( const auto &row : r.get() ) res.push_back( row );
    }        

    for ( const auto &row : res )
    {
        for ( int x : row ) std::cout << x << ' ';
        std::cout << std::endl;
    }        
}    

输出与上面相同

1 2 3 
2 2 2 
4 5 6 
6 6 6 

或者你可以写下面的方式

#include <iostream>
#include <vector>
#include <functional>

int main()
{
    std::vector<std::vector<int>> a = { { 1, 2, 3 }, { 2, 2, 2 } };
    std::vector<std::vector<int>> b = { { 4, 5, 6 }, { 6, 6, 6 } };
    std::vector<std::vector<int>> res;
    res.reserve( a.size() + b.size() );

    for ( auto &r : { std::cref( a ), std::cref( b ) } )
    {
        res.insert( res.end(), r.get().begin(), r.get().end() );    
    }        

    for ( const auto &row : res )
    {
        for ( int x : row ) std::cout << x << ' ';
        std::cout << std::endl;
    }        
}    

1 2 3 
2 2 2 
4 5 6 
6 6 6 

这就是有很多方法可以完成这项任务。