数组题目：转置矩阵

题目

标题和出处

标题：转置矩阵

出处：867. 转置矩阵

难度

1 级

题目描述

要求

给你一个二维整数数组 $\text{matrix}$，返回 $\text{matrix}$ 的转置矩阵。

矩阵的转置是指将矩阵的主对角线翻转，交换矩阵的行索引与列索引。

示例

示例 1：

输入：$\text{matrix = [[1,2,3],[4,5,6],[7,8,9]]}$
输出：$\text{[[1,4,7],[2,5,8],[3,6,9]]}$

示例 2：

输入：$\text{matrix = [[1,2,3],[4,5,6]]}$
输出：$\text{[[1,4],[2,5],[3,6]]}$

数据范围

• $\text{m}=\text{matrix.length}$
• $\text{n}=\text{matrix[i].length}$
• $\text{1}\le \text{m},\text{n}\le \text{1000}$
• $\text{1}\le \text{m}\times \text{n}\le {\text{10}}^{\text{5}}$
• ${\text{-10}}^{\text{9}}\le \text{matrix[i][j]}\le {\text{10}}^{\text{9}}$

解法

思路和算法

已知矩阵 $\text{matrix}$ 为 $m$ 行 $n$ 列。记矩阵 $\text{matrix}$ 转置后的矩阵是 ${\text{matrix}}^{\text{T}}$，则 ${\text{matrix}}^{\text{T}}$ 的行数等于 $\text{matrix}$ 的列数，${\text{matrix}}^{\text{T}}$ 的列数等于 $\text{matrix}$ 的行数，即转置后的矩阵是 $n$ 行 $m$ 列，且 ${\text{matrix}}^{\text{T}}[j][i]=\text{matrix}[i][j]$，其中 $0\le i<m$，$0\le j<n$。

创建 $n$ 行 $m$ 列的转置矩阵，然后将原矩阵中的元素赋值到转置矩阵中的对应位置即可。

代码

class Solution {
    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)$，其中 $m$ 和 $n$ 分别是矩阵 $\text{matrix}$ 的行数和列数。转置矩阵需要遍历原矩阵中的每个元素。

• 空间复杂度：$O(1)$。除了返回值以外，使用的空间复杂度是常数。