class Solution: def rotateMatrix(self, mat, n): # write code here res = [[[] for i in range(n)] for i in range(n)] for i in range(n): for j in range(n): res[j][n-i-1] = mat[i][j] return res