描述
给定一个矩阵 matrix,要求以转圈的方式打印。额外空间复杂度为 O(1)。
分析
如果一个 4x4 的矩阵 matrix 按行展开的元素顺序为:1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16。那么该举证转圈打印的顺序便是:1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10。如下图所示。
处理一个二维矩阵,往往需要把握宏观过程,而不是拘泥于繁琐的细节中。本题中的宏观过程就是先打印黑色箭头的外圈,再打印红色箭头的内圈。这样就可以抽象出一个子过程:**打印任意规模的圈。**以上图矩阵为例,只需要调用两次子过程,就可以实现按转圈方式打印这个矩阵。需要处理的细节会少很多。而如果没有把握宏观过程,而是去考虑遍历到 5 的时候怎么转弯去遍历 4,编程实现会变得繁琐很多。
还需要考虑一些特殊情况,如行数或列数为 1 的矩阵,只需要按顺序遍历行或列即可。
步骤
① 设置两个点 O1 和 O2,O1 起始坐标为矩阵左上角,O2 起始坐标为矩阵右下角。
② 打印矩阵外围一圈所有元素。
③ O1,O2 向内斜着移动一格(O1.x++,O1.y++,O2.x–,O2.y–)。如果 O1.x > O2.x,则退出。否则,打印以 O1,O2 为两个角点的一圈元素。
④ 重复 ③,直到退出。
代码
class PrintMatrixSpiralOrder {
public static void printMatrixSpiralOrder(int[][] matrix) {
int x1 = 0;
int y1 = 0;
int x2 = matrix.length - 1;
int y2 = matrix[0].length - 1;
while (x1 <= x2 && y1 <= y2) {
printCircle(matrix, x1++, y1++, x2--, y2--);
}
}
public static void printCircle(int[][] matrix, int x1, int y1, int x2, int y2) {
if (y1 == y2) { // 如果矩阵列数为 1,则直接遍历列即可
for (int x = x1; x <= x2; x++) {
System.out.print(matrix[x][y1] + " ");
}
} else if (x1 == x2) { // 如果矩阵行数为 1,则直接遍历行即可
for (int y = y1; y <= y2; y++) {
System.out.print(matrix[x1][y] + " ");
}
} else {
for (int y = y1; y < y2; y++) { // (x1,y1) -> (x1,y2-1)
System.out.print(matrix[x1][y] + " ");
}
for (int x = x1; x < x2; x++) { // (x1,y2) -> (x2-1,y2)
System.out.print(matrix[x][y2] + " ");
}
for (int y = y2; y > y1; y--) { // (x2,y2) -> (x2,y1+1)
System.out.print(matrix[x2][y] + " ");
}
for (int x = x2; x > x1; x--) { // (x2,y1) -> (x1+1,y1)
System.out.print(matrix[x][y1] + " ");
}
}
}
public static void main(String[] args) {
// 定义原矩阵
int num = 0;
int[][] a = new int[4][4];
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 4; j++) {
a[i][j] = ++num;
}
}
// 打印原矩阵
System.out.println("原矩阵:");
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 4; j++) {
System.out.print(a[i][j] + " ");
}
System.out.println();
}
// 转圈打印矩阵
System.out.print("转圈打印矩阵:");
printMatrixSpiralOrder(a);
}
out
原矩阵:
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
转圈打印矩阵:1 2 3 4 8 12 16 15 14 13 9 5 6 7 11 10