Skip to content
文档章节

48-旋转图像

题目参考: 48-旋转图像

官方题解参考: 官方参考

题目描述

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

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

py
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[[7,4,1],[8,5,2],[9,6,3]]
py
输入: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]]

下面的后两种方案我没有想到,参考官方题解。

暴力解法

创建一个新的矩阵 answer, 将原矩阵 matrix 移动元素 a(i,j) -> a(n-i-1,j)

复杂度分析

  • 时间复杂度:O()
  • 空间复杂度:O()
py
class Solution:
    def rotate(self, matrix):
      """
      Do not return anything, modify matrix in-place instead.
      """
      n = len(matrix)

      # 方案一:另外造一个矩阵存储新数据
      # 矩阵: m x n -> n x m
      # (i,j) ->(j, m-i -1)
      answer = []
      for i in range(n):
        row = [0 for x in range(n)]
        answer.append(row)

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

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

原地旋转

方案二: 矩阵位子转换操作 假设 a(i,j) 是矩阵上的一个元素,则

py
1. a(i,j) -> a(j, n-i-1)
2. a(j, n-i-1) -> a(n-i-1, n-j-1)
3. a(n-i-1, n-j-1) -> a(n-j-1, i)
4. a(n-j-1, i) -> a(i, j)

可知上面四个位置元素使 一组相关元素,形成闭环旋转

当 n 为偶数时,我们需要枚举 n^2 /4=(n/2)×(n/2) 个位置,可以将该图形分为四块,在矩阵中旋转这些块即可。

当 n 为奇数时,由于中心的元素会在旋转后回到原地,我们可以看成是 4 个一组进行旋转,只需枚举 n^2 -1 /4=(n-1/2)×(n+1/2) 个位置,然后进行旋转。

复杂度分析

  • 时间复杂度:O()
  • 空间复杂度:O(1)
py
class Solution:
    def rotate(self, matrix):
      """
      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):
            temp = matrix[i][j]

            matrix[i][j] = matrix[n-j-1][i]
            matrix[n-j-1][i] = matrix[n-i-1][n-j-1]
            matrix[n-i-1][n-j-1] = matrix[j][n-i-1]
            matrix[j][n-i-1] = temp

方案三: 用翻转代替旋转

  1. 水平上下翻转
  2. 对角翻转

复杂度分析

  • 时间复杂度:O()
  • 空间复杂度:O(1)
py
class Solution:
    def rotate(self, matrix):
      """
      Do not return anything, modify matrix in-place instead.
      """
      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]