Update 48. Rotate Image.py

Author: xulzee

48. Rotate Image.py

@@ -8,7 +8,7 @@
 
 
 class Solution:
-    def rotate(self, matrix: List[List[int]]) -> None:
+    def rotate1(self, matrix: List[List[int]]) -> None:
         """
         Do not return anything, modify matrix in-place instead.
         """
@@ -17,12 +17,33 @@ def rotate(self, matrix: List[List[int]]) -> None:
                 matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]
             matrix[i] = matrix[i][::-1]
 
+    def rotate(self, matrix: List[List[int]]) -> None:
+        start_row, start_column = 0, 0
+        end_row, end_column = len(matrix) - 1, len(matrix) - 1
+        while start_row < end_row:
+            self.rotateEdge(matrix, start_row, start_column, end_row, end_column)
+            start_row += 1
+            start_column += 1
+            end_row -= 1
+            end_column -= 1
+
+    def rotateEdge(self, matrix: list, start_row: int, start_column: int, end_row: int, end_column: int) -> None:
+        times = end_row - start_row
+        for i in range(times):
+            temp = matrix[start_row][start_column + i]
+            matrix[start_row][start_column + i] = matrix[end_row - i][start_column]
+            matrix[end_row - i][start_column] = matrix[end_row][end_column - i]
+            matrix[end_row][end_column - i] = matrix[start_row + i][end_column]
+            matrix[start_row + i][end_column] = temp
+
 
 if __name__ == '__main__':
     A = [
-        [1, 2, 3],
-        [4, 5, 6],
-        [7, 8, 9]
+        [5, 1, 9, 11],
+        [2, 4, 8, 10],
+        [13, 3, 6, 7],
+        [15, 14, 12, 16]
     ]
+    # Solution().rotate(A)
     Solution().rotate(A)
     print(A)