LeetCode 48. 旋转图像

48. 旋转图像

给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。

你必须在 原地 旋转图像，这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。

示例 1：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[[7,4,1],[8,5,2],[9,6,3]]

示例 2：

输入：matrix = [[5,1,9,11],[2,4,8,10],[13,3,6,7],[15,14,12,16]]
输出：[[15,13,2,5],[14,3,4,1],[12,6,8,9],[16,7,10,11]]

提示：

• n == matrix.length == matrix[i].length
• 1 <= n <= 20
• -1000 <= matrix[i][j] <= 1000

解法1：使用辅助数组

将图像旋转 90 度之后，对于矩阵中的第一行而言，在旋转后，它出现在倒数第一列的位置：
并且，第一行的第 x 个元素在旋转后恰好是倒数第一列的第 x 个元素。

对于矩阵中的第二行而言，在旋转后，它出现在倒数第二列的位置：
对于矩阵中的第三行和第四行同理。这样我们可以得到规律：

对于矩阵中第 i 行的第 j 个元素，在旋转后，它出现在倒数第 i 列的第 j 个位置。

我们将其翻译成代码。由于矩阵中的行列从 0 开始计数，因此对于矩阵中的元素 matrix[row][col]，在旋转后，它的新位置为 matrixnew[col][n−row−1]。

这样以来，我们使用一个与 matrix 大小相同的辅助数组 matrixnew，临时存储旋转后的结果。我们遍历 matrix 中的每一个元素，根据上述规则将该元素存放到 matrixnew 中对应的位置。在遍历完成之后，再将 matrixnew 中的结果复制到原数组中即可。

Java版：

class Solution {
    public void rotate(int[][] matrix) {
        int n = matrix.length;
        int[][] ans = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                ans[j][n - 1 - i] = matrix[i][j];
            }
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                matrix[i][j] = ans[i][j];
            }
        }
    }
}

复杂度分析

• 时间复杂度：O(n * n)，n是 数组 matrix 的长度。
• 空间复杂度：O(n * n)，n是 数组 matrix 的长度。

Python3版：

class Solution:
    def rotate(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        n = len(matrix)
        ans = [[0] * n for _ in range(n)]

        for i in range(n):
            for j in range(n):
                ans[j][n - 1 - i] = matrix[i][j]

        for i in range(n):
            for j in range(n):
                matrix[i][j] = ans[i][j]

复杂度分析

• 时间复杂度：O(n * n)，n是 列表 matrix 的长度。
• 空间复杂度：O(n * n)，n是 列表 matrix 的长度。

二维列表生成表达式：

list=[表达式 for 变量 in 范围 if 条件]

• 表达式 是包含变量的式子，作为每一个列表元素出现
• for 变量 in 范围 是for循环生成列表元素
• if 条件 是对生成的元素进行判断(可以省略)

[0]*n
list * int 意思是将数组重复 int 次并依次连接形成一个新数组

for _ in range(n)
for _ in range(n)仅将循环运行n次，等效于for i in range(n)，只不过_在后面不会用到，只是占位符，这里的_可以替换成任何符合规定的字符串。