题目

WeChatfd66e7a7f2b3c1986ee2d6355b34369e.png

解题思路

旋转90度可以看做,先transpose,再reflect。稍微画一画图就能理解。则此题关键在于transpose和reflect的写法。(详细见下方)

transpose:a[i][j] <-> a[j][i], diagonal indice 不用调换,且注意a[0][1] <->a[1][0]调换后,下一次循环要避免再次执行 a[1][0] <->a[0][1],故j应该从(i+1,n),即每次都从diagonal indice之后开始(右边倒三角)

WeChat379b5ef15e82f3b264dedb8719f8aeb0.png

reflect:每行中间对称互换,这个的规律更简单,i range (0,n), j range (0, n//2)。注意j到一半即可。

  1. class Solution:
  2. def rotate(self, matrix: List[List[int]]) -> None:
  3. """
  4. Do not return anything, modify matrix in-place instead.
  5. """
  6. self.transpose(matrix)
  7. self.reflect(matrix)
  8. # transpose
  9. def transpose(self, matrix):
  10. n = len(matrix)
  11. for i in range(n):
  12. for j in range(i+1, n):
  13. matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]
  14. def reflect(self, matrix):
  15. n = len(matrix)
  16. for i in range(n):
  17. for j in range(n//2):
  18. matrix[i][j], matrix[i][n-j-1] = matrix[i][n-j-1], matrix[i][j]