LeetCode刷题(八)

72 阅读1分钟

旋转矩阵

给你一幅由 N × N 矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 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]
]

解题思路:
先沿着对角线反转 然后整体左右翻转就欧克了
如果可以使用新的内存空间的话可以创建一个新的数组。旧的数组按行读取之后,按照列,从最后一列插入
java

class Solution {
    public void rotate(int[][] matrix) {
        int n = matrix.length;
        int math = 0;
        if(n%2==0){
            math = n/2;
        }
        else{
            math = n/2 + 1;
        }
        for(int i = 0;i<n;i++){
            for(int j =i;j<n;j++){
                int tmp = matrix[i][j]; 
                matrix[i][j]=matrix[j][i];
                matrix[j][i]=tmp;
            }
        }
        for(int i = 0;i<n;i++){
            for(int j =0;j<math;j++){
                int tmp=matrix[i][j];
                matrix[i][j]=matrix[i][n-j-1];
                matrix[i][n-j-1]=tmp;
            }
        }
    }
}

python

class Solution(object):
    def rotate(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: None Do not return anything, modify matrix in-place instead.
        """
        
        # 对角线替换
        # 左右翻转
        n = len(matrix)
        for i in range(n):
            for j in range(i,n):
                tmp = matrix[i][j] 
                matrix[i][j]=matrix[j][i]
                matrix[j][i]=tmp
        if n%2==0:
            math = n/2
        else:
            math = n/2+1
        for i in range(n):
            for j in range(math):
                tmp=matrix[i][j]
                matrix[i][j]=matrix[i][n-j-1]
                matrix[i][n-j-1]=tmp
        return matrix