给定一个矩阵 A
, 返回 A
的转置矩阵。
矩阵的转置是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。
示例 1:
输入:[[1,2,3],[4,5,6],[7,8,9]]
输出:[[1,4,7],[2,5,8],[3,6,9]]
示例 2:
输入:[[1,2,3],[4,5,6]]
输出:[[1,4],[2,5],[3,6]]
提示:
1 <= A.length <= 1000
1 <= A[0].length <= 1000
解析:
这是一个非常简单的题目,但是我还是做了好几遍才做对,实在是惭愧,我一开始只考虑的方阵,所以就在原矩阵的基础上转置,后来才发现了问题。这个题目原矩阵是row行col列,则需要构建一个col行row列的矩阵作为转置后的矩阵,然后依次赋值,就可以得到结果。
代码:
vector<vector<int>> transpose(vector<vector<int>>& A)
{
int row = A.size();
int col = 0;
//按照原题,这里可以不用判断
if (row > 0)
col = A[0].size();
else
return A;
//构建一个新矩阵
vector<vector<int>>result = vector<vector<int>>(col, vector<int>(row));
for (int i = 0; i < row; i++)
{
for (int j = 0; j < col; j++)
{
result[j][i] = A[i][j];
}
}
return result;
}