二维数组. 转置矩阵

转自：二维数组. 转置矩阵 - 知乎

二维数组. 转置矩阵

Diamon
2 人赞同了该文章

867. 转置矩阵

给你一个二维整数数组 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