867. 转置矩阵
给定一个矩阵 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 R=A.size();
// int C=A[0].size();
// int size = R*C;
// if(R!=C) return A;
// for(int i =0;i<size;i++)
// {
// if(i/C<i%C){
// int temp =A[i/C][i%C];//把原来的给temp
// A[i/C][i%C]=A[i%C][i/C];//后来的给原来
// A[i%C][i/C]=temp;//temp给后来的
// }
// }
// return A;
// }
vector<vector<int>> transpose(vector<vector<int>>& A) {
int R=A.size();
int C=A[0].size();
int size = R*C;
vector<vector<int> > res(C,vector<int>(R));//创建一个返回vector,指定转置后的行和列数
for(int i=0;i<C;i++)
{
for(int j=0;j<R;j++)
{
res[i][j]=A[j][i];
}
}
return res;
}
};