使用C ++中的向量转置2D矩阵

时间:2018-03-23 07:57:03

标签: c++ matrix vector

我试图找出使用向量的2D矩阵的转置。以下代码对我不起作用。我试图将矩阵传递给名为transpose()的函数。我能做到的其他方法是什么?如果你指出我的错误,我感激不尽。

#include <iostream>
#include <vector>
using namespace std;

void transpose(vector<vector<int> > &b)
{
    vector<vector<int> > trans_vec;

    for(int i=0;i<b.size();i++)
    {
        for(int j=0;j<b[i].size();j++)
        {
            trans_vec[j][i]=b[i][j];
        }
    }
}

int main()
{
    vector<vector<int> > v1;
    for(int i=0;i<2;i++)
    {
        vector<int> temp;
        for(int j=0;j<3;j++)
        {
            temp.push_back(i);
        }
        v1.push_back(temp);
    }
    //Display output
    for(int i=0;i<v1.size();i++)
    {
        for(int j=0;j<v1[i].size();j++)
        {
            cout<<v1[i][j];
        }
        cout<<endl;
    }
    transpose(v1);
    //display transposed matrix
    for(int i=0;i<v1.size();i++)
    {
        for(int j=0;j<v1[i].size();j++)
        {
            cout<<v1[i][j];
        }
        cout<<endl;
    }
    return 0;
}

1 个答案:

答案 0 :(得分:1)

您将保持传入的参数不变。这就是为什么你看不到变化的原因。 此外,您正在访问无效索引,因此请使用正确的大小初始化矢量。试试这个:

void transpose(vector<vector<int> > &b)
{
    if (b.size() == 0)
        return;

    vector<vector<int> > trans_vec(b[0].size(), vector<int>());

    for (int i = 0; i < b.size(); i++)
    {
        for (int j = 0; j < b[i].size(); j++)
        {
            trans_vec[j].push_back(b[i][j]);
        }
    }

    b = trans_vec;    // <--- reassign here
}

这会在第一行中将矢量初始化为正确的大小。 当然这只有作用,如果b [0] .size()== b [1] .size()== ... == b [n] .size()。

如果你不想使用push_back,你可以写:

    for (int j = 0; j < b[i].size(); j++)
    {
        if (trans_vec[j].size() != b.size())
            trans_vec[j].resize(b.size());
        trans_vec[j][i] = b[i][j];
    }

返回结果的另一种方法是在函数末尾返回trans_vec并将b作为const引用传入:

vector<vector<int> > transpose(const vector<vector<int> > &b)
{
     ...
     return trans_vec;
}