LeetCode刷题记录——第867题(转置矩阵)
程序员文章站
2024-03-22 11:03:40
...
题目描述
给定一个矩阵 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
思路分析
- 用 res 构造一个与A转置大小一致的矩阵,然后将元素写入
代码示例
class Solution(object):
def transpose(self, A):
"""
:type A: List[List[int]]
:rtype: List[List[int]]
"""
m = len(A)
n = len(A[0])
res = [[0] * m for i in range(n)]
for i in range(m):
for j in range(n):
res[j][i] = A[i][j]
return res
- 2019/3/17 晚8点,于燕园。大分法真的很难学啊,答案都不会抄的那种
上一篇: typedef & #define