Как повернуть квадратную матрицу?

Question

[PenguinVasily]

Выполняю задачу на LeetCode - 48. Rotate Image. Суть задачи заключается в том, чтобы повернуть квадратную матрицу на 90 градусов. Я написал алгоритм, который это делает:

class Solution:
    def rotate(self, matrix: List[List[int]]) -> None:        
        size = len(matrix)
        
        rotated_matrix = []
        for i in range(size):
            column = []
            for j in range(size):
                column.append(matrix[size - j - 1][i])
            rotated_matrix.append(column)
                
        matrix = rotated_matrix

Но проблема в том, что, по условию задачи, необходимо изменить входную матрицу. Когда я пытаюсь заменить её на повернутую матрицу, то ничего не выходит - тесты показывают значения входной матрицы. Как это исправить?

Comments

[PenguinVasily]

Само задание

Answer 1

[0xD34F]

class Solution:
  def rotate(self, matrix: List[List[int]]) -> None:
    length = len(matrix)
    for i in range(0, length // 2):
      for j in range(i, length - i - 1):
        matrix[i][j], \
        matrix[j][length - i - 1], \
        matrix[length - i - 1][length - j - 1], \
        matrix[length - j - 1][i] \
        = \
        matrix[length - j - 1][i], \
        matrix[i][j], \
        matrix[j][length - i - 1], \
        matrix[length - i - 1][length - j - 1]

Comments

[mayton2019]

Можно как-то отрефакторить

def rotate(self, matrix: List[List[int]]) -> None:
    m = matrix
    length = len(m)
    for i in range(0, length // 2):
      for j in range(i, length - i - 1):
        k = length - i - 1
        l = length - j - 1
        m[i][j], m[j][k], m[k][l], m[l][i] = m[l][i], m[i][j], m[j][k], m[k][l]