一、题目描述
给定一个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
`
二、代码
代码如下:
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
n = len(matrix[0])
#先转置matrix
for i in range(n):
for j in range(i):
temp = matrix[j][i]
matrix[j][i] = matrix[i][j]
matrix[i][j] = temp
#再调换每一列的位置,沿着中心对称
for i in range(n):
#此处调换列时只需要调换“一半”的列即可,因为另一半已经被调换好了
for j in range(n//2):
temp = matrix[i][j]
matrix[i][j] = matrix[i][n-1-j]
matrix[i][n - 1 - j] = temp
return matrix
三、解题思路
本题如果采用新增矩阵的形式去做可能会更容易做出来,但如果要求原地旋转的话则需要用点技巧。
本题的解题思路是先将矩阵进行转置,即 A —> AT
意思就是先将行变成列,列变成行,即第1行变成第1列,第2行变成第2列…依次类推,得到矩阵matrix的转置后,我们发现,旋转90°后的图像为矩阵matrix转置的镜像对称,对称轴就是矩阵的中心列,所以只需要将对称轴左侧的列全部沿着对称轴与对称轴右侧的列替换,就能得出结果。
需要注意的是,镜像替换时的替换次数应该为:n//2(此处的n为列数,“//”表示整除),如果替换次数还是 n 次的话,那么到最后相当于替换后又再次替换回来,矩阵并没有变化。