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:
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        m=len(A)
        n=len(A[0])
        res = m*[n*[0]]
        for i in range(n):
            for j in range(m):
                res[i][j]=A[j][i]
        return res
'''
运行错误，输入A = [[1,2,3],[4,5,6],[7,8,9]]
输出的是：[[3,6,9],[3,6,9],[3,6,9]]
'''

代码一修改：

就改变了一行，这两行生成的结果是一样的，问题出在哪我也没想明白，如果有人看到知道答案的，烦请下方留言告知。

class Solution:
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        m=len(A)
        n=len(A[0])
        #res = m*[n*[0]]
        res=[[0 for i in range(m)] for j in range(n)]
        for i in range(n):
            for j in range(m):
                res[i][j]=A[j][i]
        return res

代码三：利用高阶函数

class Solution:
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        return map(list,zip(*A))