LeetCode 867. 转置矩阵

139 阅读1分钟

题目

给定一个矩阵 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