leetcode 将一个二维矩阵进行90度旋转

import numpy as np

import  math

if __name__ == '__main__':

    def rotate(matrix):

        n = len(matrix[0])

        for i in range(math.ceil((n-1)/2)):

            for j in range(i,n-i-1):

                temp = matrix[i][j]

                matrix[i][j] = matrix[n-1-j][i]

                matrix[n-1-j][i] = matrix[n-1-i][n-1-j]

                matrix[n-1-i][n-1-j] = matrix[j][n-1-i]

                matrix[j][n-1-i] = temp

        return matrix

matrix =np.arange(1,17).reshape(4,4)

print(rotate(matrix))

如图3×3的旋转过程：

如图4×4旋转过程：