转自:二维数组. 转置矩阵 - 知乎
二维数组. 转置矩阵
Diamon
2 人赞同了该文章
给你一个二维整数数组 matrix
, 返回 matrix
的 转置矩阵 。
矩阵的 转置 是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[[1,4,7],[2,5,8],[3,6,9]]
示例 2:
输入:matrix = [[1,2,3],[4,5,6]]
输出:[[1,4],[2,5],[3,6]]
思路
- 需要生成的一个新的矩阵(二维数组),它的行数是原矩阵的列数,它的列数是原矩阵的行数
- 遍历,交换行列索引
复杂度分析
- 时间复杂度:O(mn),其中 m 和 n 分别是矩阵matrix 的行数和列数。需要遍历整个矩阵,并对转置后的矩阵进行赋值操作。
- 空间复杂度:O(mn)。(官网说是O(1),明显是错了)
Java实现
public int[][] transpose(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
int[][] t = new int[n][m];
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
// 交换行列索引
t[j][i] = matrix[i][j];
}
}
return t;
}
Python实现
思路:R*C的矩阵变成C*R的矩阵,ans[c][r]=A[r][c]
class Solution(object):
def transpose(self, A):
R, C = len(A), len(A[0])
ans = [[None] * R for _ in range(C)] #新矩阵 C行R列
for r, row in enumerate(A):
for c, val in enumerate(row):
ans[c][r] = val
return ans