轉置矩陣
阿新 • • 發佈:2020-12-12
題目描述:
給定一個矩陣 A, 返回 A 的轉置矩陣。矩陣的轉置是指將矩陣的主對角線翻轉,交換矩陣的行索引與列索引。
提示:
1 <= A.length <= 1000
1 <= A[0].length <= 1000
示例 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]]
程式碼如下:
class Solution {
public:
vector< vector<int>> transpose(vector<vector<int>>& a) {
vector<vector<int>>v;
vector<int>vec;
int H=a.size();
int L=a[0].size();
for(int i=0;i<L;i++){
for(int j=0;j<H;j++){
vec.push_back(a[j][i] );
}
v.push_back(vec);
vec.clear();
}
return v;
}
};
執行結果: