转置矩阵
给你一个二维整数数组 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]]
解题思路
一个 m * n 的矩阵,通过转置后会变成一个 n * m 的矩阵。
即定义一个 n * m 的矩阵 res,对原矩阵进行遍历,每次遍历时,使 res[j][i] = matrix[i][j],最终就会得到一个转置矩阵 res。
代码
class Solution {
public int[][] transpose(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
int[][] res = new int[n][m];
for (int i=0;i<m;i++) {
for (int j=0;j<n;j++) {
res[j][i] = matrix[i][j];
}
}
return res;
}
}
时间复杂度:因为遍历了一次矩阵,复杂度为O(mn)
空间复杂度:因为定义了一个新的矩阵,复杂度为O(mn)