48 - 旋转图像 - python + Java

93 阅读2分钟

将图像顺时针旋转 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]
]

题目要求将图像顺时针旋转90度,最后返回旋转后的结果。这里图像表示为矩阵的形式,那么返回的就是旋转90度后的矩阵。我们以示例1 来看,如果将其矩阵按行展开为 [ 1 , 2 , 3 ] , [ 4 , 5 , 6 ] , [ 7 , 8 , 9 ] [1,2,3],[4,5,6],[7,8,9] [1,2,3],[4,5,6],[7,8,9],旋转后的结果同样按行展开 [ 7 , 4 , 1 ] , [ 8 , 5 , 2 ] , [ 9 , 6 , 3 ] [7,4,1],[8,5,2],[9,6,3] [7,4,1],[8,5,2],[9,6,3],我们可以看出有如下的规律:


在这里插入图片描述

知道了规律的存在,使用zip()就可以很简单的解决问题啦~

class Solution:
    def rotate(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        # 矩阵旋转,行当列
        matrix[:] = [t[::-1] for t in zip(*matrix)]

上面的方法是一种借助Python讨巧的方法,如果是按照正常的思路来说,可以选择先转置矩阵,然后翻转每一行,最后得到的矩阵就是想要的矩阵。

public class Solution {

    public static void main(String[] args) {
        int[][] matrix = {{1,2,3}, {4,5,6}, {7,8,9}};
        rotate(matrix);

    }

    public static void rotate(int[][] matrix) {
        int n = matrix.length;
        for (int i = 0; i < n; i++) {
            for (int j = i; j < n; j++){
                int t = matrix[j][i];
                matrix[j][i] = matrix[i][j];
                matrix[i][j] = t;
            }
        }

        printMatrix(matrix);
        System.out.println("-----------------");

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n / 2; j++) {
                int tmp = matrix[i][j];
                matrix[i][j] = matrix[i][n - j - 1];
                matrix[i][n - j - 1] = tmp;
            }
        }

        printMatrix(matrix);
    }

    public static void printMatrix(int[][] matrix) {
        for (int i = 0; i != matrix.length; i++) {
            for (int j = 0; j != matrix[0].length; j++) {
                System.out.print(matrix[i][j] + " ");
            }
            System.out.println();
        }
    }
}