题目:给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。
你必须在 原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。
链接 https://leetcode.cn/problems/rotate-image
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
# 规律:列下标变为行下标,旧的行下标与新的行下标之和为n
n = len(matrix)
for i in range(n // 2):
# 一圈一圈地旋转,从外圈到内圈,即从(0,0)到(1,1)...
for j in range(i,n-i-1):
num = 0
temp = matrix[i][j]
while num < 4:
# 旋转后的位置
i,j = j,n-i-1
nextNum = matrix[i][j]
matrix[i][j] = temp
temp = nextNum
num += 1
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
# 规律:列下标变为行下标,旧的行下标与新的行下标之和为n
n = len(matrix[0])
matrix2 = [[0]*n for i in range(n)]
for i in range(n):
for j in range(n):
matrix2[j][n-i-1] = matrix[i][j]
matrix[:] = matrix2[:]
复杂度分析
时间复杂度:O(N ^ 2),其中 N 是matrix 的边长。
空间复杂度:O(N ^ 2)。我们需要使用一个和 matrix 大小相同的辅助数组。
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
n = len(matrix)
for i in range(n // 2):
for j in range((n + 1) // 2):
matrix[i][j], matrix[n - j - 1][i], matrix[n - i - 1][n - j - 1], matrix[j][n - i - 1] \
= matrix[n - j - 1][i], matrix[n - i - 1][n - j - 1], matrix[j][n - i - 1], matrix[i][j]
class Solution:
def rotate(self, matrix: List[List[int]]) -> None:
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]
复杂度分析
时间复杂度:O(N ^ 2),其中 N 是 matrix 的边长。对于每一次翻转操作,我们都需要枚举矩阵中一半的元素。
空间复杂度:O(1)。为原地翻转得到的原地旋转。
作者:LeetCode-Solution
链接:https://leetcode.cn/problems/rotate-image/solution/xuan-zhuan-tu-xiang-by-leetcode-solution-vu3m/