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 <= A.length <= 1000

1 <= A[0].length <= 1000

解題思路

class Solution:
    def transpose(self, A: [[int]]) -> [[int]]:
        res = [[0]*len(A) for i in range(len(A[0]))]
        for i in range(len(A)) :
            for j in range(len(A[i])):
                res[j][i] = A[i][j]
        return res