给定一个 n × n 的二维矩阵 matrix
表示一个图像。请你将图像顺时针旋转 90 度。
你必须在 原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。
思路:
考虑不借助辅助矩阵,通过在原矩阵中直接「原地修改」,实现空间复杂度 O(1) 的解法。
以位于矩阵四个角点的元素为例,设矩阵左上角元素 A 、右上角元素 B 、右下角元素 C 、左下角元素 D 。
旋转操作变为:
暂存 tmp=A,A←D←C←B←tmp
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
n = len(matrix)
for i in range(n // 2):
for j in range((n + 1) // 2):
tmp = matrix[i][j]
# 左下到左上
matrix[i][j] = matrix[n - 1 - j][i]
# 右下到左下
matrix[n - 1 - j][i] = matrix[n - 1 - i][n - 1 - j]
# 右上到右下
matrix[n - 1 - i][n - 1 - j] = matrix[j][n - 1 - i]
# 左上到右上
matrix[j][n - 1 - i] = tmp