48.旋转图像

·题目描述

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

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

 

示例 1：


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


输入：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]]

·解题思路——————将旋转拆分为两步计算：转置和交换

1.矩阵转置：对于第一行第一列元素对应交换。然后row+= 1，col+= 1，表示转置的矩阵缩小一圈，第一行第一列的位置更新 2.矩阵左右交换：若矩阵col为奇数，先找到中间middle（该col不需要交换），对于middle左右的两列相互交换。交换完成，左边列减一，右边列加一，表示不断向外扩展

·代码

from math import ceil

class Solution(object):
    def rotate(self, matrix):
        row_num = len(matrix)
        col_num = len(matrix[0])

        row , col = 0, 0
        while row < row_num:
            for i in range (col , col_num):
                matrix [row][i] , matrix [i][row] = matrix [i][row], matrix [row][i]

            row += 1
            col += 1

        if col_num % 2 != 0 :
            middle_num = ceil((col_num-1) / 2)
            left_num = middle_num - 1
            right_num = middle_num + 1
        else:
            right_num = ceil((col_num-1) / 2)
            left_num = (col_num-1) // 2


        while left_num >= 0:
            for i in range(row_num):
                matrix[i][left_num], matrix[i][right_num] = matrix[i][right_num], matrix[i][left_num]

            left_num -= 1
            right_num += 1

        return matrix