python3+数学思路解决旋转矩阵图像

问题描述

给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。

你必须在 原地 旋转图像，这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。

示例一：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]] 输出：[[7,4,1],[8,5,2],[9,6,3]]

示例二：

输入：matrix = [[5,1,9,11],[2,4,8,10],[13,3,6,7],[15,14,12,16]]
输出：[[15,13,2,5],[14,3,4,1],[12,6,8,9],[16,7,10,11]]

解题思路

观察图的规律，会发现旋转后，第一行变成了最后一列，第二行变成了倒数第二列，即第i行数字经过旋转对应成为（总列数-i）列的位置。根据这一思路计算出每一个位置的对应公式，遍历数组。

代码实现

import copy
class Solution:
    def rotate(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        # 复制矩阵但不会受原矩阵影响的一种创建方式
        matrix_f = copy.deepcopy(matrix)
        # 获取行、列数
        rows = len(matrix)
        cols = len(matrix[0])
        # 遍历矩阵
        # 这里转换一下思考位置，因为需要直接改变原矩阵，因此从第一行开始遍历时，应该想到从后面开始旋转，即最后一行是第一列，倒数第二行是第二列    
        # 原矩阵的第i行第j列经过旋转应该变为原矩阵的第rows-i-j行第i列
        for i in range(0,rows):
            for j in range(0,cols):        
                matrix[i][j] = matrix_f[rows-1-j][i]
        return matrix