【程序员面试金典】01.07.旋转矩阵

旋转矩阵

给你一幅由 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]
]

题目解法

要将矩阵旋转90°，最简单的做法就是一层一层进行旋转。对每一层执行环状旋转：将上边移到右边，右边移到下边，下边移到左边，左边移到上边。

如何交换4条边呢？一种做法是把上面复制到一个数组中，然后将左边移到上边，下边移到右边，等等。但这需要占用O(N)的内存空间，实际上没有必要。

更好的做法是按索引一个一个进行交换，具体做法如下。

for i = 0 to n
    temp = top[i]
    top[i] = left[i]
    left[i] = bottom[i]
    bottom[i] = right[i]
    right[i] = temp

从最外面一层开始逐渐向里，在每一层上执行上述交换。另外，也可以从内层开始，逐层向外。

算法实现（JAVA）

public class Solution {
    public void rotate(int[][] matrix) {
        if (matrix.length == 0 || matrix.length != matrix[0].length) {
            return;
        }

        int n = matrix.length;
        for (int i = 0; i < n / 2; i++) { // 当前层
            // 当前层长度
            int last = n - 1 - i;
            for (int j = i; j < last; j++) {
                // 当前层第几个元素
                int offset = j - i;

                // 存储上边(第i层第j列)
                int top = matrix[i][j];

                // 从左边(当前层即为最左边元素的列，行为当前层从下往上数第offset个)移到上边
                matrix[i][j] = matrix[last - offset][i];

                // 下边(当前层最后一行，第last -offset列)移到左边
                matrix[last - offset][i] = matrix[last][last - offset];

                // 右边移到下边
                matrix[last][last - offset] = matrix[j][last];

                // 上边移到右边
                matrix[j][last] = top;
            }
        }
    }

    public static void main(String[] args) {
        int[][] matrix =
                {
                        {5, 1, 9, 11},
                        {2, 4, 8, 10},
                        {13, 3, 6, 7},
                        {15, 14, 12, 16}
                };
        Solution solution = new Solution();


        // [[15, 13, 2, 5]
        // [14, 3, 4, 1]
        // [12, 6, 8, 9]
        // [16, 7, 10, 11]]
        solution.rotate(matrix);
        for (int[] m : matrix) {
            System.out.println(Arrays.toString(m));
        }
    }
}

[1] 盖尔.拉克曼.麦克道尔.程序员面试金典（第6版）[M].北京：人民邮电出版社，2019.9：166-167
[2] leetcode.面试题 01.07. 旋转矩阵