leetcode 867. 轉置矩陣
阿新 • • 發佈:2018-11-26
給定一個矩陣 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): """ :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))