旋转矩阵
给你一幅由 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. 旋转矩阵