题目
难度:★☆☆☆☆
类型:几何、二维数组、数学
给定一个矩阵 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
解答
转置前矩阵的维度是r=len(A), c=len(A[0]),转置后矩阵的维度应该交换,首先我们构建转置后的矩阵,并填充所有值为空,然后遍历A矩阵中的每一个点,把它放在B上对应的位置即可:B[j][i]=A[i][j]。
class Solution:
def transpose(self, A):
"""
:param A: List[List[int]]
:return: List[List[int]]
"""
B = [[None for _ in range(len(A))] for _ in range(len(A[0]))]
for i in range(len(A)):
for j in range(len(A[0])):
B[j][i] = A[i][j]
return B
在python中有zip方法,可以实现快速的矩阵转置:
class Solution:
def transpose(self, A):
"""
:param A: List[List[int]]
:return: List[List[int]]
"""
return list(zip(*A))
如有疑问或建议,欢迎评论区留言~