二维数组. 转置矩阵

最新推荐文章于 2024-07-10 21:32:09 发布

gf771115

最新推荐文章于 2024-07-10 21:32:09 发布

阅读量1.7k

点赞数

分类专栏： JAVA

JAVA 专栏收录该内容

41 篇文章

订阅专栏

本文详细介绍了如何通过算法实现矩阵的转置，并提供了Java和Python两种语言的具体实现代码。矩阵转置是将矩阵的行变为列，列变为行的过程。文章通过示例展示了转置前后矩阵的变化，并分析了算法的时间和空间复杂度。

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

二维数组. 转置矩阵

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