我试图找出使用向量的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;
}
答案 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;
}