题目
给你一个二维整数数组 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]]
解题思路
原数组的行和列,正好就是新数组的列和行。对换即可
示例代码
public int[][] transpose(int[][] matrix) {
int m = matrix.length, n = matrix[0].length;
int[][] transposed = new int[n][m];
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
transposed[j][i] = matrix[i][j];
}
}
return transposed;
}
复杂度分析
时间复杂度:O(mn),其中 mm 和 nn 分别是矩阵的行数和列数。需要遍历整个矩阵,并对转置后的矩阵进行赋值操作。
空间复杂度:O(1)。除了返回值以外,额外使用的空间为常数。
题目链接:
867. 转置矩阵