【模拟-BM99 顺时针旋转矩阵】

题目

BM99 顺时针旋转矩阵
描述
有一个NxN整数矩阵，请编写一个算法，将矩阵顺时针旋转90度。

给定一个NxN的矩阵，和矩阵的阶数N,请返回旋转后的NxN矩阵。

分析

模拟，写几个样例，分析一下新矩阵元素下标与原矩阵元素下标的关系，可以发现ans[i][j] = mat[n-1-j][i]

代码

class Solution:
    def rotateMatrix(self , mat: List[List[int]], n: int) -> List[List[int]]:
        # write code here
        ans = [[0]*n for i in range(n)]
        for i in range(n):
            for j in range(n):
                ans[i][j] = mat[n-1-j][i]
        return ans