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:
    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))