1. 程式人生 > >【Leetcode】 867. 轉置矩陣

【Leetcode】 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. 1 <= A.length <= 1000
  2. 1 <= A[0].length <= 1000

這題只需要像題目說的一樣,把行當成列,列當成行,模擬一下就可以了,就算沒學過線性代數都可以做出來!

程式碼如下:

class Solution {
    public int[][] transpose(int[][] A) {
        int[][] ans=new int[A[0].length][A.length];
        for(int i=0;i<A[0].length;i++){
            for(int j=0;j<A.length;j++){
                ans[i][j]=A[j][i];
            }
        }
        return ans;
    }
}