查看原文
其他

漫画:如何螺旋遍历二维数组?(修订版)

脚本之家 2022-04-23

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 },
                { 1112131415 },
                { 1617181920 }
        };
        int[][] matrix2 = {
                { 1,  2,  3 },
                { 4,  5,  6 },
                { 7,  8,  9 },
                { 101112 },
                { 131415 }
        };
        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

使得同一条边不会被重复遍历到。

推荐阅读:

漫画:如何螺旋遍历二维数组?
漫画:什么是 “黑天鹅事件” ?
漫画:什么是 “跳表” ?

每日打卡赢积分兑换书籍入口

👇🏻👇🏻👇🏻


您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存