题目描述
给定一个矩阵 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
源代码
class Solution {
public:
vector<vector<int>> transpose(vector<vector<int>>& A){
int row=A.size();
int col=A[0].size();
vector<vector<int>> temp(col,vector<int>(row,0));
for (int i=0;i<row;i++){
for (int j=0;j<col;j++){
temp[j][i]=A[i][j];
}
}
return temp;
}
};
注释
二维向量的输入问题:
不像二维数组那样,可以直接对 arr[i][j] 进行行循环赋值。在vector<vector>中,因为vector是一个容器,最外层的vector容器中放着更小的vector,而里层的vector里面放的是int型的数字。所以我们首先要对里层的vector容器赋值,然后再把里层的vector作为元素插入到外层的vector中。
行列的表示:
e.g.: vector<vector> array; 则 行 row=array.size( ); 列 col=array[0].size( );