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

思路1:

類C。建立一個新矩陣,逐一填入即可。

程式碼1:

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

分析:

時間複雜度O(n^ 2), 空間複雜度O(n^2)

思路2:

Pythonic思路。使用map函式和zip函式。

程式碼2:

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

分析:

使用內建高階函式較快。