漫画:如何螺旋遍历二维数组?(修订版)
The following article is from 程序员小灰 Author 小灰
来源:程序员小灰(ID:chengxuyuanxiaohui)
在周一发布的漫画当中,小灰忽略了一个小问题:
当二维数组的最内层只有一行或一列时,螺旋遍历有可能重复访问。因此必须在第3和第4个小循环中加上额外的条件限制。
今天出了一个修订版,修正了这个缺陷。
非常感谢大家的指正。
————— 第二天 —————
什么意思呢?我们来举个例子,给定下面这样一个二维数组:
我们需要从左上角的元素1开始,按照顺时针进行螺旋遍历,一直遍历完所有的元素,遍历的路径就像下图一样:
经过这样的遍历,返回的元素结果如下:
1,2,3,4,5,10,15,20,19,18,17,16,11,6,7,8,9,14,13,12
————————————
第1层
从左到右遍历“上边”:
从上到下遍历“右边”:
从右到左遍历“下边”:
从下到上遍历“左边”:
第2层
从左到右遍历“上边”:
从上到下遍历“右边”:
从右到左遍历“下边”:
从下到上遍历“左边”:
第3层
从左到右遍历“上边”:
从上到下遍历“右边”:
从右到左遍历“下边”:
第三层的“左边”已无需遍历,二维数组到此遍历完毕。
public class SpiralOrder {
public static List<Integer> spiralOrder(int[][] matrix) {
List<Integer> list = new ArrayList<Integer>();
//当二维数组是空或任何一个维度是0,直接返回
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return list;
}
//m是矩阵的行数
int m = matrix.length;
//n是矩阵的列数
int n = matrix[0].length;
//二维数组的层数,取决于行和列的较小值
int size = (Math.min(m, n)+1)/2;
//大循环,从外向内逐层遍历矩阵
for(int i=0; i<size; i++) {
//从左到右遍历“上边”
for (int j=i; j<n-i; j++) {
list.add(matrix[i][j]);
}
//从上到下遍历“右边”
for (int j=i+1; j<m-i; j++) {
list.add(matrix[j][(n-1)-i]);
}
//从右到左遍历“下边”
for (int j=i+1; j<n-i && (m-1)-i>i; j++) {
list.add(matrix[(m-1)-i][(n-1)-j]);
}
//从下到上遍历“左边”
for (int j=i+1; j<m-1-i && i<(n-1)-i; j++) {
list.add(matrix[(m-1)-j][i]);
}
}
return list;
}
public static void main(String[] args) {
int[][] matrix = {
{ 1, 2, 3, 4, 5 },
{ 6, 7, 8, 9, 10 },
{ 11, 12, 13, 14, 15 },
{ 16, 17, 18, 19, 20 }
};
int[][] matrix2 = {
{ 1, 2, 3 },
{ 4, 5, 6 },
{ 7, 8, 9 },
{ 10, 11, 12 },
{ 13, 14, 15 }
};
List<Integer> resultList1 = spiralOrder(matrix);
System.out.println(Arrays.toString(resultList1.toArray()));
List<Integer> resultList2 = spiralOrder(matrix2);
System.out.println(Arrays.toString(resultList2.toArray()));
}
}
在上面的代码中,一个大循环当中包含了4个小循环。大循环控制了每一层的遍历,4个小循环分别实现了同一层上边、右边、下边,左边的遍历。
当遍历到最内层时,4个小循环并不会全都执行,比如测试代码中matrix2的最内层就只有一列,此时只需要遍历“上边”和“右边”。
这种情况下,遍历“下边”和“左边”的小循环必须加上了额外的条件限制:
(m-1)-i>i
i<(n-1)-i
使得同一条边不会被重复遍历到。
推荐阅读:
每日打卡赢积分兑换书籍入口
👇🏻👇🏻👇🏻