1:题目描述 (力扣):
给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。
你必须在 原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。
2:解题思路
方法一:使用辅助数组
使用题目中的示例二作为例子:
[5 1 9 11] [2 4 8 10] [13 3 6 7] [15 14 12 16]
将图像顺时针旋转90°后,第一行出现在倒数第一列
[5 1 9 11]
[0 0 0 0]
[0 0 0 0]
[0 0 0 0]
进行旋转后:
[0 0 0 5]
[0 0 0 1]
[0 0 0 9]
[0 0 0 11]
第一行的第x个元素在旋转后恰好在倒数第一列的第x个元素。
以此类推:第二行进行旋转后在倒数第二列,第三行与第四行同理。
可以得出:第i行的第j个元素,进行旋转后,出现在倒数第i列的第j个元素。
因此对于矩阵中的元素matrix[row][rol],在旋转后,它的新位置为matrix[rol][len(matrix)-row-1]。
这样一来,使用一个与matrix大小相同的辅助数组new_matrix临时存储旋转后的结果,旋转结束后,再将new_matrix中的结果复制到原数组中。
代码展示:
import copy
class Solution:
def rotate(self, matrix):
if not len(matrix) or not len(matrix[0]):
return None
n = len(matrix)
# 使用深拷贝
new_matrix = copy.deepcopy(matrix)
# 使用列表生成式,位置的数据用0填充
# new_matrix = [[0] * n for _ in range(n)]
row = 0
for item in matrix:
j = (len(matrix)-1)- row
for i in range(len(item)):
new_matrix[i][j] = item[i]
row += 1
matrix[:] = new_matrix
return matrix
方法二:用翻转代替旋转
先水平翻转,在进行对角线翻转。还是以题目中的示例二为例子:
[5 1 9 11]
[2 4 8 10]
[13 3 6 7]
[15 14 12 16]
先将其进行水平翻转,得到:
[15 14 12 16]
[13 3 6 7]
[2 4 8 10]
[5 1 9 11]
再根据主对角线翻转,得到:
[15 13 2 5]
[14 3 4 1]
[12 6 8 9]
[16 7 10 11]
对于水平翻转而言,我们只需要枚举矩阵上半部分的元素与矩阵下半部分的元素进行交换,即
matrix[row][rol]--------->matrix[len(matrix)-row-1][rol]
对于主对角线翻转而言,我们只需要枚举矩阵对角线左侧的元素与右侧的元素进行交换,即
matrix[row][rol]--------->matrix[rol][row],此处的row,rol是针对水平翻转后的矩阵
把他们联立可得到:
matrix[row][rol]--------->matrix[len(matrix)-row-1][rol]--------->matrix[rol][len(matrix)-row-1]
代码展示:
class Solution:
def rotate(self, matrix):
n = len(matrix)
# 水平翻转
for i in range(n//2): # 只枚举上半部分的元素
for j in range(n):
matrix[i][j], matrix[n-i-1][j] = matrix[n-i-1][j], matrix[i][j]
# 对角线翻转
for i in range(n):
# 只需要将对角线左侧的元素与对角线右侧的元素进行翻转
for j in range(i): # 只枚举对角线左侧的元素
matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]
return matrix